Class LSUNDataset

Inheritance Relationships

Base Type

Class Documentation

class LSUNDataset : public mindspore::dataset::Dataset

A source dataset for reading LSUN datast.

Public Functions

LSUNDataset(const std::vector<char> &dataset_dir, const std::vector<char> &usage, const std::vector<std::vector<char>> &classes, bool decode, const std::shared_ptr<Sampler> &sampler, const std::shared_ptr<DatasetCache> &cache)

Constructor of LSUNDataset.

Parameters
  • dataset_dir[in] Path to the root directory that contains the dataset.

  • usage[in] Dataset splits of LSUN, can be train, valid, test or all.

  • classes[in] Classes list to load.

  • decode[in] Decode the images after reading.

  • sampler[in] Shared pointer to a sampler object used to choose samples from the dataset.

  • cache[in] Tensor cache to use.

LSUNDataset(const std::vector<char> &dataset_dir, const std::vector<char> &usage, const std::vector<std::vector<char>> &classes, bool decode, const Sampler *sampler, const std::shared_ptr<DatasetCache> &cache)

Constructor of LSUNDataset.

Parameters
  • dataset_dir[in] Path to the root directory that contains the dataset.

  • usage[in] Dataset splits of LSUN, can be train, valid, test or all.

  • classes[in] Classes list to load.

  • decode[in] Decode the images after reading.

  • sampler[in] Sampler object used to choose samples from the dataset.

  • cache[in] Tensor cache to use.

LSUNDataset(const std::vector<char> &dataset_dir, const std::vector<char> &usage, const std::vector<std::vector<char>> &classes, bool decode, const std::reference_wrapper<Sampler> sampler, const std::shared_ptr<DatasetCache> &cache)

Constructor of LSUNDataset.

Parameters
  • dataset_dir[in] Path to the root directory that contains the dataset.

  • usage[in] Dataset splits of LSUN, can be train, valid, test or all.

  • classes[in] Classes list to load.

  • decode[in] Decode the images after reading.

  • sampler[in] Sampler object used to choose samples from the dataset.

  • cache[in] Tensor cache to use.

~LSUNDataset() override = default

Destructor of LSUNDataset.