一种基于改进YOLOv8n的小目标检测方法

    公开(公告)号:CN116895007A

    公开(公告)日:2023-10-17

    申请号:CN202310877467.3

    申请日:2023-07-18

    Abstract: 本发明提出一种基于改进YOLOv8n的小目标检测方法。方法具体步骤如下:(1)构建小目标数据集;(2)引入过渡下采样模块替代YOLOv8n模型浅层下采样操作,减少了输入图像由于连续下采样导致的小目标信息丢失;(3)在YOLOv8n模型中分别插入上采样模块和下采样模块,减少了较浅层网络中由于下采样导致的小目标信息丢失;(4)采用融合了改进SwinTransformer‑V2模块的改进C3模块替换YOLOv8n模型中部分C2f模块,加强了小目标信息在主干特征提取网络深层部分的关注度;(5)利用(1)中构建的数据集对改进的YOLOv8n模型进行训练、验证、评价。本发明在减少模型少量参数量、满足实时检测要求的情况下,增加了小目标检测的平均精度均值,减少了误检漏检情况的发生。

Patent Agency Ranking