pysegcnn.core.dataset

Custom dataset classes compliant to the PyTorch standard.

Each custom dataset should inherit from torch.utils.data.Dataset to benefit from the torch.utils.data.DataLoader class, which implements helpful utilities during model training.

For any kind of image-like dataset, inherit the ImageDataset class to create your custom dataset.

License

Copyright (c) 2020 Daniel Frisinghelli

This source code is licensed under the GNU General Public License v3.

See the LICENSE file in the repository’s root directory.

Classes

Cloud95Dataset(root_dir[, use_bands, …])

Class for the `Cloud-95`_ dataset by `Mohajerani et al.

ImageDataset(root_dir[, use_bands, …])

Base class for multispectral image data.

ProSnowDataset(root_dir[, use_bands, …])

Class for the ProSnow datasets.

ProSnowGarmisch(root_dir[, use_bands, …])

Class for the ProSnow Garmisch dataset.

ProSnowObergurgl(root_dir[, use_bands, …])

Class for the ProSnow Obergurgl dataset.

SparcsDataset(root_dir[, use_bands, …])

Class for the `Sparcs`_ dataset.

StandardEoDataset(root_dir[, use_bands, …])

Base class for standard Earth Observation style datasets.

SupportedDatasets(value)

Names and corresponding classes of the implemented datasets.