Class ConvertColor
Defined in File vision.h
Inheritance Relationships
Base Type
public mindspore::dataset::TensorTransform
(Class TensorTransform)
Class Documentation
-
class ConvertColor : public mindspore::dataset::TensorTransform
Change the color space of the image.
Public Functions
-
explicit ConvertColor(ConvertMode convert_mode)
Constructor.
- Parameters
convert_mode – [in] The mode of image channel conversion.
Example/* dataset is an instance of Dataset object */ dataset = dataset->Map({std::make_shared<vision::Decode>(), std::make_shared<vision::ConvertColor>(ConvertMode::COLOR_BGR2RGB)}, // operations {"image"}); // input columns
-
~ConvertColor() = default
Destructor.
-
explicit ConvertColor(ConvertMode convert_mode)