首页 > 其他分享 >YOLOv5+DecoupleHead解耦头(YOLOx)

YOLOv5+DecoupleHead解耦头(YOLOx)

时间:2024-07-13 11:57:13浏览次数:12  
标签:YOLOv5 耦头 源站 YOLOx 防盗链 ms 外链 图片

一、解耦头原理
在目标检测中,分类任务和回归任务之间的冲突是一个众所周知的问题。因此,用于分类和定位的解耦头被广泛应用于大多数一级和二级探测器。但是,由于YOLO系列的主干和特征金字塔(如FPN, PAN)不断演化,它们的检测头仍然是耦合的。

从下表可以看出,头耦合时端到端属性降低了4.2% AP,而头解耦时,端到端属性降低了0.8% AP。因此,我们将YOLO检测头替换为简化解耦头。具体来说,它包含一个1 × 1的conv层来降低信道维数,然后是两个平行的分支,分别有两个3 × 3的conv层。我们在表2中报告了V100上批处理=1时的推断时间,简化解耦头带来了额外的1.1 ms (11.6 ms vs . 10.5 ms)。

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-d4sJH45h-1720425938808)(https://i-blog.csdnimg.cn/direct/60972f0cdc0c40f8b219ff5781313481.png)]
上图中,YOLOv3头与解耦头之间的区别示意图。对于FPN的每一层特征,我们首先采用1×1的conv层将特征通道减少到256,然后添加两个并行分支,每个分支有2个3 × 3conv层,分别用于分类和回归任务。在回归分支上添加IoU分支。

二、解耦头对收敛速度的影响
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-OyXXBH6v-1720425938811)(https://i-blog.csdnimg.cn/direct/4f609b4e2ff649f4b612ff2b3c50849c.png)]
解耦头的收敛速度比YOLOv3头快得多,最终取得了较好的收敛效果。

三、解耦头对精度的影响
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来

标签:YOLOv5,耦头,源站,YOLOx,防盗链,ms,外链,图片
From: https://blog.csdn.net/qq_41920323/article/details/140262816

相关文章

  • 玩一玩yolov5 自己训练模型识别马克杯
    python虚拟环境搭建condacreate-nyolopython==3.8yolov5下载gitclonehttps://github.com/ultralytics/yolov5cdyolov5activateyolopipinstall-rrequirements.txt准备数据集官方介绍:https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data建立文件......
  • 施工现场不戴安全帽抓拍 YOLOv5
    施工现场不戴安全帽抓拍利用现场已经有的摄像头,施工现场不戴安全帽抓拍运用机器视觉边缘计算和神经网络深度学习算法,对现场进出口、作业区域等人员违规行为识别、分析与预警提醒,施工现场不戴安全帽抓拍并把警报截屏和视频储存到后台。此外,施工现场不戴安全帽抓拍还可以识别现场......
  • C#实现控制台传参调用YoloV5进行人体识别
    一、项目地址 二、代码解析3、自定义人体信息类  该类用于在控制台打印人体位置信息///<summary>///人体信息///</summary>publicclassBodyInfo{publicintLeft{get;set;}=0;publicintTop{get;set;}=0;......
  • 掌握yolov5【一】
    文章目录前言一、安装miniconda二、安装pytorch三、下载yolov5源码总结前言yolov5介绍YOLOv5(YouOnlyLookOnceversion5)是一个广受欢迎的实时目标检测模型,由Ultralytics发布。它基于YOLO系列模型,旨在提供更高效和准确的目标检测解决方案。YOLOv5的设计目标是简化......
  • autoware.universe源码略读(3.4)--perception:tensorrt_yolox
    autoware.universe源码略读3.4--perception:tensorrt_yoloxOverview结构体预定义TrtYoloXpreprocessdoInferencefeedforwardfeedforwardAndDecodedecodeOutputstensorrt_yolox_nodeyolox_single_image_inference_node总结Overview可以看到,其实在最新版本的autoware......
  • yoloV5实现红外海上船舶分类
     介绍Pytorch使用yoloV5实现红外海上船舶分类硬件配置实验在一台配备以下规格的机器上进行:-处理器:[i5-10200H]-GPU:[GTX1650]-内存:[16GB]软件配置-操作系统:[Windows10]-深度学习框架:PyTorch[1.10.2]-CUDA版本:[11.3]-cuDNN版本:[11.6]模型使用的是yol......
  • 主干网络篇 | YOLOv5/v7 更换主干网络之 ShuffleNetv2 | 高效CNN架构设计的实用指南(2)
    主干网络篇|YOLOv5/v7更换主干网络之ShuffleNetv2|高效CNN架构设计的实用指南概述YOLOv5和YOLOv7是目前主流的轻量级目标检测模型,在速度和精度方面取得了良好的平衡。然而,传统的YOLOv5/v7模型使用FPN和CSPNet等结构作为主干网络,在移动设备和嵌入式系统等资源受限的场景......
  • YOLOX算法实现血细胞检测
    原文:YOLOX算法实现血细胞检测-知乎(zhihu.com)目标检测一直是计算机视觉中比较热门的研究领域。本文将使用一个非常酷且有用的数据集来实现YOLOX算法,这些数据集具有潜在的真实应用场景。问题陈述 数据来源于医疗相关数据集,目的是解决血细胞检测问题。任务是通过显微......
  • 【YOLOv5/v7改进系列】改进池化层为ASPP
    一、导言AtrousSpatialPyramidPooling(ASPP)模块是一种用于多尺度特征提取的创新技术,旨在提升深度学习模型在语义图像分割任务中的表现。ASPP模块通过在不同的采样率下应用空洞卷积,可以捕获不同大小的对象以及图像的上下文信息,从而增强模型在处理不同尺度物体时的鲁棒性......
  • 口罩佩戴检测系统 YOLOv5
    正确的佩戴口罩对现阶段有效减低人员之间感染新型冠状病毒具有重要意义。基于YOLOv5在图像识别检测领域的优异性能,本文研究基于基于YOLOv5的口罩佩自动戴检测方法。首先从网络和真实生活中中寻找并采集不同场景人群口罩佩戴的图片约500张并自建数据集,然后利用YOLOv5模型框架,修改其......