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