1、首先6.0版本中有针对yolov5.pt转yolov5.onnx的export.py脚本文件,在文件中主要注意下面这个问题: 设置opset=11或者12都可以 2、转换为onnx文件后,可以通过netron地址查看一下模型的结构。 Netronnetron.app/ yolov5s的模型结构中,网络将【1,3,80,80,85】、【1,3,40,40,85】、【1,3,20,20,85】...
使用ATC Yolov5中的onnx转换om模型可以成功,但使用通过yolov5的github提供的与训练模型yolov5s.pt并使用自己的训练集进行训练得到的pt文件转成的onnx模型在转换成om模型时失败。 初步分析: 应该可以排除版本和环境问题,因为能够跑通示例,不知道是不是模型转换的时候出错了。 软件版本: -- 原始模型框架 (Pytorch) ...