ToMode¶
- class ToMode(mode='CHW', *, order=None)[source]¶
Change input data to a target mode. For example, most transforms use HWC mode image, while the neural network might use CHW mode input tensor.
- Parameters
mode – output mode of input. Default: “CHW”
order – the same with
VisionTransform