Model_zoo模型导出设置与模型转换设置不匹配
收藏回复举报
Model_zoo模型导出设置与模型转换设置不匹配
t('forum.solved') 已解决
发表于2024-03-14 14:28:07
0 查看

问题描述:

在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

本帖最后由 匿名用户2024/03/20 09:54:30 编辑

我要发帖子