brails.processors.FoundationClassifier.csail_segmentation_tool.csail_seg.dataset module

class brails.processors.FoundationClassifier.csail_segmentation_tool.csail_seg.dataset.BaseDataset(odgt, opt, **kwargs)

Bases: Dataset

img_transform(img)

parse_input_list(odgt, max_sample=-1, start_idx=-1, end_idx=-1)

round2nearest_multiple(x, p)

segm_transform(segm)

class brails.processors.FoundationClassifier.csail_segmentation_tool.csail_seg.dataset.TestDataset(odgt, opt, **kwargs)

Bases: BaseDataset

class brails.processors.FoundationClassifier.csail_segmentation_tool.csail_seg.dataset.TrainDataset(root_dataset, odgt, opt, batch_per_gpu=1, **kwargs)

Bases: BaseDataset

class brails.processors.FoundationClassifier.csail_segmentation_tool.csail_seg.dataset.ValDataset(root_dataset, odgt, opt, **kwargs)

Bases: BaseDataset

brails.processors.FoundationClassifier.csail_segmentation_tool.csail_seg.dataset.imresize(im, size, interp='bilinear')