Class UniformAugment
Defined in File vision.h
Inheritance Relationships
Base Type
public mindspore::dataset::TensorTransform
(Class TensorTransform)
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.
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.
-
explicit UniformAugment(const std::vector<TensorTransform*> &transforms, int32_t num_ops = 2)