data module

class cellseg.data.DataProcessor(image_dir, label_dir, target_size=(512, 512), image_suffix='tif')[source]

Bases: Generic[torch.utils.data.dataset.T_co]

transform(image, mask)[source]
Parameters

  • image – Image to be transformed.

  • mask – Label/Mask to be transformed.

Returns

Transformed images that have been randomly cropped, flipped, and resized to the target size.