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

link

1、复制这些

2、utils-loss,这里加

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

4、先对CoputerLoss类做修改

5、把那一行替换成这个

6、修改

7、iou_ration是超参,可以调,如果小目标比较多的话,这个值可以低一些,从而增加nwd损失的比例

8、还有一个部分要改


9、这个box也要改

相关推荐
2501_9418372615 小时前
基于YOLOv8的19种鱼类目标检测与分类系统——鱼类市场物种识别研究
yolo·目标检测·分类
AAD5558889916 小时前
牛肝菌目标检测:基于YOLOv8-CFPT-P2345模型的创新实现与应用_1
人工智能·yolo·目标检测
音沐mu.17 小时前
【50】背包数据集(有v5/v8模型)/YOLO背包检测
yolo·目标检测·数据集·背包检测·背包数据集
OLOLOadsd12317 小时前
激光设备目标检测 - 基于YOLOv5-HGNetV2的高精度检测模型实现_1
人工智能·yolo·目标检测
Ryan老房18 小时前
视频标注新方法-从视频到帧的智能转换
人工智能·yolo·目标检测·ai·目标跟踪·视频
极智视界18 小时前
无人机场景 - 目标检测数据集 - 环岛路况车辆检测数据集下载
yolo·目标检测·数据集·voc·coco·算法训练·无人机场景环岛路况车辆检测
Liue6123123118 小时前
基于YOLOv5的柑橘类叶片识别任务实现与优化
yolo
duyinbi751718 小时前
基于YOLOv10n的人脸口罩检测模型_结合FocalModulation改进技术_提升口罩检测精度与效率_1
yolo
Liue6123123119 小时前
YOLOv8-CSMHSA 狮子检测与识别模型训练与评估_1
yolo
棒棒的皮皮20 小时前
【深度学习】YOLO学习资源之官方文档&Darknet文档
深度学习·学习·yolo·计算机视觉