torch_geometric.datasets.Teeth3DS

class Teeth3DS(root: str, split: str = 'Teeth3DS', train: bool = True, num_samples: int = 30000, transform: Optional[Callable] = None, pre_transform: Optional[Callable] = None, force_reload: bool = False)[source]

Bases: InMemoryDataset

The Teeth3DS+ dataset from the “An Extended Benchmark for Intra-oral 3D Scans Analysis” paper.

This dataset is the first comprehensive public benchmark designed to advance the field of intra-oral 3D scan analysis developed as part of the 3DTeethSeg 2022 and 3DTeethLand 2024 MICCAI challenges, aiming to drive research in teeth identification, segmentation, labeling, 3D modeling, and dental landmark identification. The dataset includes at least 1,800 intra-oral scans (containing 23,999 annotated teeth) collected from 900 patients, covering both upper and lower jaws separately.

Parameters:

  • root (str) – Root directory where the dataset should be saved.

  • split (str) – The split name (one of "Teeth3DS", "3DTeethSeg22_challenge" or "3DTeethLand_challenge").

  • train (bool, optional) – If True, loads the training dataset, otherwise the test dataset. (default: True)

  • num_samples (int, optional) – Number of points to sample from each mesh. (default: 30000)

  • 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)

  • force_reload (bool, optional) – Whether to re-process the dataset. (default: False)