locpix.img_processing.data_loading.dataset.ImgDataset#
- class ImgDataset(input_root, files, transform, train=False, mean=0, std=1)[source]#
Bases:
Dataset
Pytorch dataset for the SMLM data represented as images
Attributes:
- Parameters:
input_root (string) – Directory containing the SMLM data and masks
files (list) – List of the files to include from the directory in this dataset
transform (dictionary) – Transforms to apply to the dataset
- __init__(input_root, files, transform, train=False, mean=0, std=1)[source]#
- Parameters:
input_root (string) – Directory containing the SMLM data and masks
files (list) – List of the files to include from the directory in this dataset
transform (dictionary) – Transforms to apply to the dataset
Methods
__init__
(input_root, files, transform[, ...])- param input_root:
Directory containing the SMLM data and masks