Datasets Module#

Datasets module.

landmarker.datasets.get_cepha_heatmap_datasets(path_dir, transform=None, sigma=1, kaggle=True, junior=False, single_dataset=False, **kwargs)[source]#

Returns a HeatmapDataset with the ISBI 2015 cephalogram challenge dataset. The dataset is split into train, test1 and test2. The same approach as in “CephaNN: A Multi-Head Attention Network for Cephalometric Landmark Detection” - JIAHOONG QIAN et al. is used.

Parameters:

  • path_dir (str) – The path to the directory where the dataset should be stored.

  • transform (Optional[Callable], optional) – A transformation to apply to the images and heatmaps. Defaults to None.

  • sigma (int, optional) – The sigma value for the gaussian kernel. Defaults to 1.

  • kaggle (bool, optional) – Whether to use the kaggle dataset. Defaults to True.

  • junior (bool, optional) – Whether to use the junior or senior annotator. Defaults to False.

  • single_dataset (bool, optional) – Whether to return a single dataset with all images and landmarks. Defaults to False.

  • **kwargs – Additional keyword arguments for the HeatmapDataset.

Return type:

HeatmapDataset | tuple[HeatmapDataset, HeatmapDataset, HeatmapDataset]

landmarker.datasets.get_cepha_landmark_datasets(path_dir, transform=None, store_imgs=True, dim_img=None, kaggle=False, junior=False, single_dataset=False)[source]#

Returns a LandmarkDataset objects with the CEPH dataset, a combination of the ISBI 2014 & 2015 challenges. The dataset is split into train, test1 and test2. The same approach as in “CephaNN: A Multi-Head Attention Network for Cephalometric Landmark Detection” - JIAHOONG QIAN

et al. is used.

Parameters:

  • path_dir (str) –

  • transform (Callable | None) –

Return type:

LandmarkDataset | tuple[LandmarkDataset, LandmarkDataset, LandmarkDataset]