本文主要是介绍YOLOV7改进-针对小目标的NWD(损失函数),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
link
1、复制这些
2、utils-loss,这里加
3、把这几行复制到utiils的loss.py
4、先对CoputerLoss类做修改
5、把那一行替换成这个
6、修改
7、iou_ration是超参,可以调,如果小目标比较多的话,这个值可以低一些,从而增加nwd损失的比例
8、还有一个部分要改
9、这个box也要改
这篇关于YOLOV7改进-针对小目标的NWD(损失函数)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!