Mentions légales du service

Skip to content

NIFTIFolderDataset Added

SILVA RINCON Santiago Smith requested to merge feature/254-sp15-item12-create-nifti-dataloader-for-loading-medical-images into develop

Changelog

  • Solving #254 (closed)
  • Submodule for medical data loaders added in fedbiomed.common.data.datasets
  • Added import to __all__ in fedbiomed.common.data for simplicity
  • Added unit testing in test_medical_datasets.py
  • Added documentation

Usage

As a standard PyTorch Dataset.

from fedbiomed.common.data import NIFTIFolderDataset

dataset = NIFTIFolderDataset('~/path_to_dataset')
img, label = dataset[0]
n_images = len(dataset)

Requires

  • Add dependency for medical imaging loading itk using conda install -c conda-forge itk (issue to be opened after approval)
Edited by SILVA RINCON Santiago Smith

Merge request reports