Class HWC2CHW

Defined in File vision_lite.h

Inheritance Relationships

Base Type

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

Class Documentation

class HWC2CHW : public mindspore::dataset::TensorTransform 

Transpose the input image; shape (H, W, C) to shape (C, H, W).

Public Functions

HWC2CHW()

Constructor.

Example

/* dataset is an instance of Dataset object */
dataset = dataset->Map({std::make_shared<vision::Decode>(),
                        std::make_shared<vision::HWC2CHW>()}, // operations
                       {"image"});                            // input columns

~HWC2CHW() override = default: Destructor.