brails.processors.FoundationClassifier.npid.npid_datasets.folder module

class brails.processors.FoundationClassifier.npid.npid_datasets.folder.CombinedMaskDataset(other_data_path=None, csv_root_folder=None, data_csv=None, transform=None, mask_images=True, attribute=None)

Bases: Dataset

: Slightly hackish way to train on masks quick

loader(path)

class brails.processors.FoundationClassifier.npid.npid_datasets.folder.ImageFolderInstance(root: str | ~pathlib.Path, transform: ~typing.Callable | None = None, target_transform: ~typing.Callable | None = None, loader: ~typing.Callable[[str], ~typing.Any] = <function default_loader>, is_valid_file: ~typing.Callable[[str], bool] | None = None, allow_empty: bool = False)

Bases: ImageFolder

: Folder datasets which returns the index of the image as well:

class brails.processors.FoundationClassifier.npid.npid_datasets.folder.MaskFolderInstance(*args, **kwargs)

Bases: ImageFolder

: Slightly hackish way to train on masks quick