摘要
本发明涉及图像目标检测领域,具体涉及一种基于YOLOv7‑tiny算法的轻量化目标检测方法,步骤是:首先,基于城市道路车辆流动场景构建数据集,并对其进行图像预处理;其次,采用轻量级卷积构建多尺度特征提取模块和多尺度特征融合模块,通过构建新的上采样模块获得质量更好的特征图;然后进行训练,得到改进YOLOv7‑tiny网络模型;最后将待检测图像输入改进后的检测模型中进行检测,在得到不同尺度下的检测结果后,将其融合在一起,得到最终的检测结果。经实验验证,该方法可以有效减少检测模型的参数和计算量,同时保持检测精度基本稳定,从而提升实时性和计算效率,满足无人机在目标检测和跟踪任务中的需求。
技术关键词
轻量化无人机
卷积模块
采样模块
多尺度特征提取
通道注意力机制
城市道路
网络
图像
上采样
特征提取模块
数据
特征点
场景
车辆
度量
编码
算法