torch_geometric.datasets.FAUST
- class FAUST(root: str, train: bool = True, transform: Optional[Callable] = None, pre_transform: Optional[Callable] = None, pre_filter: Optional[Callable] = None, force_reload: bool = False)[source]
Bases:
InMemoryDataset
The FAUST humans dataset from the “FAUST: Dataset and Evaluation for 3D Mesh Registration” paper, containing 100 watertight meshes representing 10 different poses for 10 different subjects.
Note
Data objects hold mesh faces instead of edge indices. To convert the mesh to a graph, use the
torch_geometric.transforms.FaceToEdge
aspre_transform
. To convert the mesh to a point cloud, use thetorch_geometric.transforms.SamplePoints
astransform
to sample a fixed number of points on the mesh faces according to their face area.- Parameters:
root (str) – Root directory where the dataset should be saved.
train (bool, optional) – If
True
, loads the training dataset, otherwise the test dataset. (default:True
)transform (callable, optional) – A function/transform that takes in an
torch_geometric.data.Data
object and returns a transformed version. The data object will be transformed before every access. (default:None
)pre_transform (callable, optional) – A function/transform that takes in an
torch_geometric.data.Data
object and returns a transformed version. The data object will be transformed before being saved to disk. (default:None
)pre_filter (callable, optional) – A function that takes in an
torch_geometric.data.Data
object and returns a boolean value, indicating whether the data object should be included in the final dataset. (default:None
)force_reload (bool, optional) – Whether to re-process the dataset. (default:
False
)
STATS:
#graphs
#nodes
#edges
#features
#classes
100
6,890
41,328
3
10