tirank.Imageprocessing.ImageDataset

class tirank.Imageprocessing.ImageDataset(*args: Any, **kwargs: Any)[source]

Bases: Dataset

PyTorch Dataset for loading image tiles (patches).

Parameters:

images_array (np.ndarray) – A stack of image tiles from crop_images.
transform (callable, optional) – A torchvision transform to apply to each image. Defaults to None.

static collate_fn(batch)[source]

Static collate function to stack images into a batch.

Parameters:: batch (list) – A list of image tensors.
Returns:: A stacked batch of images.
Return type:: torch.Tensor