YOLOV7改进-针对小目标的NWD(损失函数)

link
1、复制这些
YOLOV7改进-针对小目标的NWD(损失函数)_第1张图片
2、utils-loss,这里加
YOLOV7改进-针对小目标的NWD(损失函数)_第2张图片

3、把这几行复制到utiils的loss.py

YOLOV7改进-针对小目标的NWD(损失函数)_第3张图片
4、先对CoputerLoss类做修改
YOLOV7改进-针对小目标的NWD(损失函数)_第4张图片
5、把那一行替换成这个
YOLOV7改进-针对小目标的NWD(损失函数)_第5张图片
YOLOV7改进-针对小目标的NWD(损失函数)_第6张图片
6、修改
YOLOV7改进-针对小目标的NWD(损失函数)_第7张图片
YOLOV7改进-针对小目标的NWD(损失函数)_第8张图片
7、iou_ration是超参,可以调,如果小目标比较多的话,这个值可以低一些,从而增加nwd损失的比例

8、还有一个部分要改
YOLOV7改进-针对小目标的NWD(损失函数)_第9张图片
YOLOV7改进-针对小目标的NWD(损失函数)_第10张图片
YOLOV7改进-针对小目标的NWD(损失函数)_第11张图片
9、这个box也要改
YOLOV7改进-针对小目标的NWD(损失函数)_第12张图片

你可能感兴趣的:(yolov7,YOLO)