Class RGBA2BGR

Defined in File vision.h

Inheritance Relationships

Base Type

public mindspore::dataset::TensorTransform (Class TensorTransform)

Class Documentation

class RGBA2BGR : public mindspore::dataset::TensorTransform 

Change the format of input tensor from 4-channel RGBA to 3-channel BGR.

Public Functions

RGBA2BGR()

Constructor.

Example

/* Define operations */
auto decode_op = vision::Decode();
auto rgb2bgr_op = vision::RGBA2BGR();

/* dataset is an instance of Dataset object */
dataset = dataset->Map({decode_op, rgb2bgr_op},  // operations
                       {"image"});               // input columns

~RGBA2BGR() override = default: Destructor.