前言
最近跑了一个YOLOv7的模型,在导出模型的时候遇见了一个bug,分享一下解决过程。
先交代一下bug的发生情景:将已经训练好的.pt文件,用YOLOv7自带的export脚本导出成onnx文件, 在转出的时候就报了如下的错误:
经过调查,发现类似的情况不多见,于是我就看了脚本的源代码,发现参数里有这样的细节:
因为模型训练的时候使用了NMS,因此我就在导出的命令里加上了这个参数。(这里就不贴源命令啦,每个人项目细节不一样,以免造成误导。
重新试了一下,问题解决!
总结
一个小小的bug解决过程,希望可以帮到大家!如果有用麻烦大家点个赞~欢迎提出意见~