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

link

1、复制这些

2、utils-loss,这里加

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

4、先对CoputerLoss类做修改

5、把那一行替换成这个

6、修改

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

8、还有一个部分要改


9、这个box也要改

相关推荐
点我头像干啥2 小时前
YOLOv8模型剪枝实战:DepGraph(依赖图)方法详解
yolo·目标跟踪·剪枝
要努力啊啊啊6 小时前
YOLOv3 训练与推理流程详解-结合真实的数据样例进行模拟
人工智能·yolo·机器学习·计算机视觉·目标跟踪
FL16238631297 小时前
基于YOLOv11+PP-OCRv5深度学习的智能车牌检测与识别系统python源码+pytorch模型+评估指标曲线+精美GUI界面
yolo
codegarfield7 小时前
YOLO11中的C3K2模块
人工智能·深度学习·yolo·c3k2
要努力啊啊啊7 小时前
YOLOv2 中非极大值抑制(NMS)机制详解与实现
人工智能·深度学习·yolo·计算机视觉·目标跟踪
从零开始学习人工智能3 天前
多模型协同:基于 SAM 分割 + YOLO 检测 + ResNet 分类的工业开关状态实时监控方案
人工智能·yolo·分类
强盛小灵通专卖员3 天前
基于YOLOv12的电力高空作业安全检测:为电力作业“保驾护航”,告别安全隐患!
人工智能·深度学习·安全·yolo·核心期刊·计算机期刊
飞翔的佩奇3 天前
【完整源码+数据集+部署教程】石材实例分割系统源码和数据集:改进yolo11-CA-HSFPN
python·yolo·计算机视觉·毕业设计·数据集·yolo11·石材实例分割系统
飞翔的佩奇3 天前
【完整源码+数据集+部署教程】安检爆炸物检测系统源码和数据集:改进yolo11-REPVGGOREPA
python·yolo·计算机视觉·毕业设计·数据集·yolo11·安检爆炸物检测
音沐mu.3 天前
【20】番茄叶片病害数据集(有v5/v8模型)/YOLO番茄叶片病害检测
人工智能·yolo·目标检测·机器学习·计算机视觉·番茄叶片病害检测·番茄叶片病害数据集