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

link

1、复制这些

2、utils-loss,这里加

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

4、先对CoputerLoss类做修改

5、把那一行替换成这个

6、修改

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

8、还有一个部分要改


9、这个box也要改

相关推荐
赵药师12 小时前
Cityscape数据集转YOLO
人工智能·深度学习·yolo
我材不敲代码15 小时前
YOLOv4 核心内容详解
yolo
动物园猫15 小时前
火焰与烟雾目标检测数据集分享(适用于YOLO系列深度学习分类检测任务)
深度学习·yolo·目标检测
动物园猫16 小时前
驾驶中的疲劳状态识别数据集分享(适用于YOLO系列深度学习分类检测任务)
深度学习·yolo·分类
羊羊小栈18 小时前
基于「YOLO目标检测 + 多模态AI分析」的木材缺陷智能检测分析预警系统
人工智能·yolo·目标检测
码上掘金19 小时前
基于YOLO和大语言模型的农田杂草智能检测系统(代码、数据集、模型和论文)
人工智能·yolo·语言模型
这张生成的图像能检测吗19 小时前
(论文速读)基于改进 YOLOv8 的轻量化车辆检测与跟踪方法
yolo·计算机视觉·目标追踪
深度学习lover20 小时前
<数据集>yolo 货车识别<目标检测>
人工智能·python·yolo·目标检测·计算机视觉·货车识别
guo_xiao_xiao_1 天前
YOLOv11野生动物园大型猫科动物目标检测数据集-8075张-Animal-detection-yolov8-1
人工智能·yolo·目标检测
ZPC82102 天前
YOLO + 3D 空间定位 + 抓取姿态 完整方案
yolo·3d