Class UniformAugment

Inheritance Relationships

Base Type

Class Documentation

class UniformAugment : public mindspore::dataset::TensorTransform

Randomly perform transformations, as selected from input transform list, on the input tensor.

Public Functions

explicit UniformAugment(const std::vector<TensorTransform*> &transforms, int32_t num_ops = 2)

Constructor.

Parameters
  • transforms[in] Raw pointer to vector of TensorTransform operations.

  • num_ops[in] An integer representing the number of operations to be selected and applied.

explicit UniformAugment(const std::vector<std::shared_ptr<TensorTransform>> &transforms, int32_t num_ops = 2)

Constructor.

Parameters
  • transforms[in] Smart pointer to vector of TensorTransform operations.

  • num_ops[in] An integer representing the number of operations to be selected and applied.

explicit UniformAugment(const std::vector<std::reference_wrapper<TensorTransform>> &transforms, int32_t num_ops = 2)

Constructor.

Parameters
  • transforms[in] Object pointer to vector of TensorTransform operations.

  • num_ops[in] An integer representing the number of operations to be selected and applied.

~UniformAugment() = default

Destructor.