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

link

1、复制这些

2、utils-loss,这里加

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

4、先对CoputerLoss类做修改

5、把那一行替换成这个

6、修改

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

8、还有一个部分要改


9、这个box也要改

相关推荐
LeonDL1683 小时前
YOLOv8 在单片机上的几种部署方案
人工智能·python·单片机·嵌入式硬件·深度学习·yolo·yolov8 在单片机上的部署
LeonDL1683 小时前
YOLOv8 在单片机上部署的缺点和应对方案
python·单片机·嵌入式硬件·深度学习·yolo·yolov8在单片机上的缺点·yolov8 在单片机上的优化
王了了哇5 小时前
【YOLO(txt)格式转VOC(xml)格式数据集】以及【制作VOC格式数据集 】
xml·python·yolo
www_pp_12 小时前
# YOLOv4:目标检测的全新突破
yolo·目标检测·目标跟踪
2201_7549184116 小时前
YOLOV3 深度解析:目标检测的高效利器
yolo·目标检测·目标跟踪
白熊1881 天前
【计算机视觉】论文精读《基于改进YOLOv3的火灾检测与识别》
人工智能·yolo·计算机视觉
白熊1881 天前
【图像生成大模型】Step-Video-T2V:下一代文本到视频生成技术
人工智能·opencv·yolo·计算机视觉·大模型·音视频
yidaqiqi3 天前
[目标检测] YOLO系列算法讲解
算法·yolo·目标检测
hao_wujing3 天前
YOLOv8在单目向下多车辆目标检测中的应用
人工智能·yolo·目标检测
Dneccc3 天前
YOLO+UI(C#)开发
yolo