CerebNet.data_loader.dataset¶

class CerebNet.data_loader.dataset.CerebDataset(dataset_path, cfg, transforms, load_aux_data)[source]¶

Class for loading aseg file with augmentations (transforms)

Methods

get_subject_names

class CerebNet.data_loader.dataset.SubjectDataset(img_org, brain_seg, patch_size, slice_thickness, primary_slice)[source]¶

Single subject loader to load and prepare slices for network process.

Attributes

Returns the active plane

Methods

`locate_mask_bbox`(mask)	Find the largest connected component of the mask.
`set_plane`(plane)	Set the active plane.

roi¶: alias of dict[Literal[‘source_shape’, ‘offsets’, ‘target_shape’], tuple[int, …]]

locate_mask_bbox(mask)[source]¶

Find the largest connected component of the mask.

class CerebNet.data_loader.dataset.TestLoader(subject_path, data_cfg, transforms=None)[source]¶

Methods