我是靠谱客的博主 自由紫菜,最近开发中收集的这篇文章主要介绍(2022.8)解决Can‘t create layer“XXX” of type “NonMaxSuppression“ in function‘cv:: dnn: : dnn4_v20211220前言总结,觉得挺不错的,现在分享给大家,希望可以做个参考。
概述
前言
最近跑了一个YOLOv7的模型,在导出模型的时候遇见了一个bug,分享一下解决过程。
先交代一下bug的发生情景:将已经训练好的.pt文件,用YOLOv7自带的export脚本导出成onnx文件, 在转出的时候就报了如下的错误:
经过调查,发现类似的情况不多见,于是我就看了脚本的源代码,发现参数里有这样的细节:
因为模型训练的时候使用了NMS,因此我就在导出的命令里加上了这个参数。(这里就不贴源命令啦,每个人项目细节不一样,以免造成误导。
重新试了一下,问题解决!
总结
一个小小的bug解决过程,希望可以帮到大家!如果有用麻烦大家点个赞~欢迎提出意见~
最后
以上就是自由紫菜为你收集整理的(2022.8)解决Can‘t create layer“XXX” of type “NonMaxSuppression“ in function‘cv:: dnn: : dnn4_v20211220前言总结的全部内容,希望文章能够帮你解决(2022.8)解决Can‘t create layer“XXX” of type “NonMaxSuppression“ in function‘cv:: dnn: : dnn4_v20211220前言总结所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复