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

link

1、复制这些

2、utils-loss,这里加

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

4、先对CoputerLoss类做修改

5、把那一行替换成这个

6、修改

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

8、还有一个部分要改


9、这个box也要改

相关推荐
paopao_wu5 小时前
目标检测YOLO[04]:跑通最简单的YOLO模型训练
人工智能·yolo·目标检测
Coding茶水间9 小时前
基于深度学习的路面坑洞检测系统演示与介绍(YOLOv12/v11/v8/v5模型+Pyqt5界面+训练代码+数据集)
图像处理·人工智能·深度学习·yolo·目标检测·计算机视觉
懷淰メ20 小时前
python3GUI--【AI加持】基于PyQt5+YOLOv8+DeepSeek的智能球体检测系统:(详细介绍)
yolo·目标检测·计算机视觉·pyqt·检测系统·deepseek·球体检测
AI即插即用2 天前
即插即用系列 | CVPR 2025 WPFormer:用于表面缺陷检测的查询式Transformer
人工智能·深度学习·yolo·目标检测·cnn·视觉检测·transformer
AI即插即用2 天前
即插即用系列 | 2025 MambaNeXt-YOLO 炸裂登场!YOLO 激吻 Mamba,打造实时检测新霸主
人工智能·pytorch·深度学习·yolo·目标检测·计算机视觉·视觉检测
shayudiandian2 天前
YOLOv8目标检测项目实战(从训练到部署)
人工智能·yolo·目标检测
Hcoco_me3 天前
YOLO目标检测学习路线图
学习·yolo·目标检测
dotphoenix3 天前
在wsl ubuntu下安装,训练,验证,导出,部署YOLO的完整例子
yolo
paopao_wu4 天前
目标检测YOLO[03]:推理入门
人工智能·yolo·目标检测
深度学习lover5 天前
<项目代码>yolo遥感航拍船舶识别<目标检测>
人工智能·python·yolo·目标检测·计算机视觉·遥感船舶识别