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

link

1、复制这些

2、utils-loss,这里加

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

4、先对CoputerLoss类做修改

5、把那一行替换成这个

6、修改

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

8、还有一个部分要改


9、这个box也要改

相关推荐
思绪无限4 小时前
YOLOv5至YOLOv12升级:条形码二维码检测系统的设计与实现(完整代码+界面+数据集项目)
深度学习·yolo·目标检测·条形码二维码检测·yolov12·yolo全家桶
格林威7 小时前
如何用 eBPF 监控 GigE Vision 相机网络性能
网络·人工智能·数码相机·yolo·计算机视觉·视觉检测·工业相机
思绪无限17 小时前
YOLOv5至YOLOv12升级:植物叶片病害识别系统的设计与实现(完整代码+界面+数据集项目)
深度学习·yolo·目标检测·yolov12·yolo全家桶·植物叶片病害检测
白羊by17 小时前
YOLOv1~v11 全版本核心演进总览
深度学习·算法·yolo
QQ6765800821 小时前
智慧工厂之扬尘识别 铲车装载识别 工程重型机械识别 磁铁识别 深度学习YOLO格式图像识别第10435期
人工智能·深度学习·yolo·扬尘识别·铲车装载·工程重型机械·磁铁识别
思绪无限1 天前
YOLOv5至YOLOv12升级:行人跌倒检测系统的设计与实现(完整代码+界面+数据集项目)
深度学习·yolo·目标检测·yolov12·yolo全家桶·行人跌倒检测系统
思绪无限1 天前
YOLOv5至YOLOv12升级:行人车辆检测与计数识别系统的设计与实现(完整代码+界面+数据集项目)
人工智能·深度学习·yolo·目标检测·yolov12·yolo全家桶·行人车辆检测与计数
思绪无限1 天前
YOLOv5至YOLOv12升级:个人防具检测系统的设计与实现(完整代码+界面+数据集项目)
深度学习·yolo·目标检测·yolov12·yolo全家桶·个人防具检测
音沐mu.1 天前
【68】无人机视角行人数据集(有v5/v8模型)/YOLO无人机视角行人检测
yolo·目标检测·目标检测数据集·行人检测·无人机视角行人数据集·无人机视角行人检测·行人数据集
A7bert7771 天前
【YOLOv8部署至RDK X5】模型训练→转换bin→Sunrise 5部署
c++·人工智能·python·深度学习·yolo·机器学习