base

Description

Basic dataset model.

The abstract BaseDataset handle provides a Sequence, which yields a torch.Tensor tuple of (input image, ground truth) upon a call to __getitem__. The transformations from image to tensor data can be changed.

Classes

BaseDataset	Abstract base class for tuple datasets with storage location.
DataTriple	Tuple of train/test/validation datasets (w/ automatic splitting if necessary).
DatasetSplit	Types of dataset splits.

Functions

add_gaussian_peak(mask_np, centroid, ...[, ...])	Add a peak to the heatmap mask_np as a non-normalized gaussian at centroid.
cross_validation_splits(train_val_data, ...)	Split dataset it into num_splits and collect tuples of (rest, split).