Class RandomColorAdjust
Defined in File vision.h
Inheritance Relationships
Base Type
public mindspore::dataset::TensorTransform
(Class TensorTransform)
Class Documentation
-
class RandomColorAdjust : public mindspore::dataset::TensorTransform
Randomly adjust the brightness, contrast, saturation, and hue of the input image.
Public Functions
-
explicit RandomColorAdjust(const std::vector<float> &brightness = {1.0, 1.0}, const std::vector<float> &contrast = {1.0, 1.0}, const std::vector<float> &saturation = {1.0, 1.0}, const std::vector<float> &hue = {0.0, 0.0})
Constructor.
- Parameters
brightness – [in] Brightness adjustment factor. Must be a vector of one or two values if it is a vector of two values it needs to be in the form of [min, max] (Default={1, 1}).
contrast – [in] Contrast adjustment factor. Must be a vector of one or two values if it is a vector of two values, it needs to be in the form of [min, max] (Default={1, 1}).
saturation – [in] Saturation adjustment factor. Must be a vector of one or two values if it is a vector of two values, it needs to be in the form of [min, max] (Default={1, 1}).
hue – [in] Hue adjustment factor. Must be a vector of one or two values if it is a vector of two values, it must be in the form of [min, max] where -0.5 <= min <= max <= 0.5 (Default={0, 0}).
Example/* Define operations */ auto decode_op = vision::Decode(); auto random_color_adjust_op = vision::RandomColorAdjust({1.0, 5.0}, {10.0, 20.0}, {40.0, 40.0}); /* dataset is an instance of Dataset object */ dataset = dataset->Map({decode_op, random_color_adjust_op}, // operations {"image"}); // input columns
-
~RandomColorAdjust() = default
Destructor.
-
explicit RandomColorAdjust(const std::vector<float> &brightness = {1.0, 1.0}, const std::vector<float> &contrast = {1.0, 1.0}, const std::vector<float> &saturation = {1.0, 1.0}, const std::vector<float> &hue = {0.0, 0.0})