Class WIDERFaceDataset

Inheritance Relationships

Base Type

Class Documentation

class WIDERFaceDataset : public mindspore::dataset::Dataset

A source dataset for reading and parsing WIDERFace dataset.

Public Functions

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

Constructor of WIDERFaceDataset.

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

  • usage[in] The type of data to be read, can be “train”, “test”, “valid” or “all”. “all” will read samples from “train” and “valid”.

  • 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.

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

Constructor of WIDERFaceDataset.

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

  • usage[in] The type of data to be read, can be “train”, “test”, “valid” or “all”. “all” will read samples from “train” and “valid”.

  • decode[in] Decode the images after reading.

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

  • cache[in] Tensor cache to use.

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

Constructor of WIDERFaceDataset.

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

  • usage[in] The type of data to be read, can be “train”, “test”, “valid” or “all”. “all” will read samples from “train” and “valid”.

  • decode[in] Decode the images after reading.

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

  • cache[in] Tensor cache to use.

~WIDERFaceDataset() override = default

Destructor of WIDERFaceDataset.