공부/TIL(을 빙자한 기타)
Supported for default input shape(NCHW or NHWC) order for each DL frameworks
빛나는나무
2023. 6. 19. 19:15
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