Class TrivialAugmentWide

Inheritance Relationships

Base Type

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.

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

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
~TrivialAugmentWide() override = default

Destructor.