Supported for default input shape(NCHW or NHWC) order for each DL frameworks

Supported for default input shape(NCHW or NHWC) order for each DL frameworks

N: Batch size

C: Channel of input

H: Height of input

W: Width

 

ONNX: NCHW

TensorRT: NCHW, NHWC

Tensorflow Lite:  NCHW, NHWC

OpenVino: NCHW

Tensorflow Keras: NCHW, NHWC

Tensorflow saved_model(pb): NCHW, NHWC