tumourkit.segmentation.hovernet.dataloader.train_loader.FileLoader

class tumourkit.segmentation.hovernet.dataloader.train_loader.FileLoader(file_list, with_type=False, input_shape=None, mask_shape=None, mode='train', setup_augmentor=True, target_gen=None)

Bases: Dataset

Data Loader. Loads images from a file list and performs augmentation with the albumentation library. After augmentation, horizontal and vertical maps are generated.

Args:

file_list: list of filenames to load input_shape: shape of the input [h,w] - defined in config.py mask_shape: shape of the output [h,w] - defined in config.py mode: ‘train’ or ‘valid’

Methods

setup_augmentor