now_2023.datasets.MRIDataset

class now_2023.datasets.MRIDataset(img_dir: str, data_df: DataFrame, transform: Callable | None = None)[source]
__init__(img_dir: str, data_df: DataFrame, transform: Callable | None = None)[source]
Parameters:

  • img_dir (str) – Path to the CAPS directory containing preprocessed images.

  • data_df (DataFrame) – Metadata of the population. Columns include participant_id, session_id, and diagnosis.

  • transform (Callable, optional) – List of transforms applied on-the-fly, chained with torchvision.transforms.Compose.

Methods

__init__(img_dir, data_df[, transform])

param img_dir:

Path to the CAPS directory containing preprocessed images.

eval()

train()