问题描述:
在yolov5目录下,在模型导出时,将模型的输入设置为(input_data = ms.numpy.zeros([config.batch_size, config.testing_shape[0], config.testing_shape[1], 3], ms.int8)),即为NHWC格式的输入。但为什么在ATC模型转换命令中的参数设置为(--input_shape="actual_input_1:1,12,320,320"),即为NCHW格式的?在进行ATC模型转换时输入数据的shape难道不是应当保持一致么?
模型转换:cid:link_0
模型导出:https://gitee.com/mindspore/models/blob/master/official/cv/YOLOv5/export.py
问题描述:
在yolov5目录下,在模型导出时,将模型的输入设置为(input_data = ms.numpy.zeros([config.batch_size, config.testing_shape[0], config.testing_shape[1], 3], ms.int8)),即为NHWC格式的输入。但为什么在ATC模型转换命令中的参数设置为(--input_shape="actual_input_1:1,12,320,320"),即为NCHW格式的?在进行ATC模型转换时输入数据的shape难道不是应当保持一致么?
模型转换:cid:link_0
模型导出:https://gitee.com/mindspore/models/blob/master/official/cv/YOLOv5/export.py