YOLOv5目标检测优化点（添加小目标头检测）_yolov5识别不到小目标

作者：我家小花儿 | 2024-05-16 13:42:53

踩

yolov5识别不到小目标

1、前言

在检测过程中，碰到小的目标物体，采用YOLOv5模型，难以检测到小目标，因此本文提出，增加小目标检测头，从浅层特征图学习到小目标的特征信息后，与深层特征图拼接，从而使深层网络也能够学习到小目标的特征信息，让网络更加关注小目标的检测，提高目标检测效果。唯一缺点是随着网络层数增多，带来计算量的增加导致检测推理速度减慢。

2、如何计算参数

H₂=(H₁-K+2P)/S+1

其中，H₁代表输入宽度，K代表卷积核个数，P代表零填充，S代表步长。

Conv：卷积，一般来说宽高会降低一遍（主要看参数）。
Upsample: 上采样，宽高升高一倍。
Concat：拼接，宽高一致情况下，维度相加。

3、YOLOv5小目标改进-1

3.1、结构文件

yolov5l.yaml 源文件

# YOLOv5        本文内容由网友自发贡献，转载请注明出处：https://www.wpsshop.cn/w/我家小花儿/article/detail/579029

推荐阅读

相关标签

YOLOv5目标检测优化点（添加小目标头检测）_yolov5识别不到小目标

文章目录

1、前言

2、如何计算参数

3、YOLOv5小目标改进-1

3.1、结构文件