但是这时候把自己的模型拿过去跑的话,会闪退。。。。
这个东西折磨了我n天,后来科学上网之后看到了这篇链接
因为我的模型是剪枝过的,对这玩意也一窍不通,最开始看到的时候就没试。。昨天突然想到下载的包里有个doc文件夹,里面有4张代码图片,我最开始还合计为什么要放这几张图片,昨天突然想通了可能要按照这个图片里的东西改一下,结果照里面改完之后代码都跑不了了。。。(刚发现人家README里面写的很清楚,只不过我没看。。。我是nt)
今天在修bug的时候突然想到可能训练的时候那个东西不用改,导出的时候要改,然后试了一下,然后。。。成功了。。。
总结一下就是训练好模型之后,用exporter.py导出onnx之前把C2f和Detect按照图片里的代码改一下再导出就完事了。。亲测剪枝后的模型也可用