基于分组快速空间金字塔池化的轻量化YOLO模型目标检测方法
Abstract:
本发明公开一种基于分组快速空间金字塔池化的轻量化YOLO模型,包括如下步骤:步骤1)利用跨阶段局部网络和分组卷积,构建分组快速空间金字塔池化结构GFSPP;步骤2)利用通道空间意力模块(CBAM)和GFSPP结构,构建基于分组快速空间金字塔池化结构的主干网络模块GFSPP‑Backbone;步骤3)利用幻影卷积结构,构建轻型颈部网络模块Ghost‑Slim‑Neck;步骤4)按照主干网络模块GFSPP‑Backbone、颈部网络模块Ghost‑Slim‑Neck、检测头网络模块的顺序,将步骤2)、步骤3)以及检测头网络模块依次串联成为轻量化目标检测模型GFSPP‑YOLO。本发明的方法能够大幅度提升主干网络模块特征提取能力的同时,显著降低颈部网络模块的计算量。
Patent Agency Ranking
0/0