Class TrivialAugmentWide
Defined in File vision.h
Inheritance Relationships
Base Type
public mindspore::dataset::TensorTransform
(Class TensorTransform)
Class Documentation
-
class TrivialAugmentWide : public mindspore::dataset::TensorTransform
Dataset-independent data-augmentation with TrivialAugment Wide.
Public Functions
-
explicit TrivialAugmentWide(int32_t num_magnitude_bins = 31, InterpolationMode interpolation = InterpolationMode::kNearestNeighbour, const std::vector<uint8_t> &fill_value = {0, 0, 0})
Constructor.
- Example
/* Define operations */ auto decode_op = vision::Decode(); auto trivial_augment_wide_op = vision::TrivialAugmentWide(); /* dataset is an instance of Dataset object */ dataset = dataset->Map({decode_op, trivial_augment_wide_op}, // operations {"image"}); // input columns
- Parameters
num_magnitude_bins – [in] The number of different magnitude values. Default: 31.
interpolation – [in] An enum for the mode of interpolation. Default: InterpolationMode::kNearestNeighbour.
InterpolationMode::kNearestNeighbour, Interpolation method is nearest-neighbor interpolation.
InterpolationMode::kLinear, Interpolation method is blinear interpolation.
InterpolationMode::kCubic, Interpolation method is bicubic interpolation.
InterpolationMode::kArea, Interpolation method is pixel area interpolation.
fill_value – [in] A vector representing the pixel intensity of the borders. Default: {0, 0, 0}.
-
~TrivialAugmentWide() = default
Destructor.
-
explicit TrivialAugmentWide(int32_t num_magnitude_bins = 31, InterpolationMode interpolation = InterpolationMode::kNearestNeighbour, const std::vector<uint8_t> &fill_value = {0, 0, 0})