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

link

1、复制这些

2、utils-loss,这里加

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

4、先对CoputerLoss类做修改

5、把那一行替换成这个

6、修改

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

8、还有一个部分要改


9、这个box也要改

相关推荐
王锋(oxwangfeng)1 小时前
YOLOWorld 实现开集障碍物检测
yolo
喵叔哟2 小时前
02-YOLO-v8-v9-v10工程差异对比
人工智能·yolo·机器学习
2501_9413331017 小时前
数字识别与检测_YOLOv3_C3k2改进模型解析
人工智能·yolo·目标跟踪
xsc-xyc21 小时前
RuntimeError: Dataset ‘/data.yaml‘ error ❌ ‘_lz
人工智能·深度学习·yolo·计算机视觉·视觉检测
张3蜂1 天前
我希望做的是识别身份证正反面,我需要标注多少张图片?
yolo
AAD555888991 天前
YOLOv8-MAN-Faster电容器缺陷检测:七类组件识别与分类系统
yolo·分类·数据挖掘
AI浩1 天前
YOLO-IOD:面向实时增量目标检测
yolo·目标检测·目标跟踪
wfeqhfxz25887821 天前
YOLOv8-BiFPN鸟巢目标检测与识别实战教程
yolo·目标检测·目标跟踪
Katecat996631 天前
基于YOLOv8和MAFPN的骆驼目标检测系统实现
人工智能·yolo·目标检测
ZCXZ12385296a1 天前
YOLOv8_HSPAN_机器人视觉系统中的球体目标检测与追踪_1
yolo·目标检测·机器人