改进的Tiny YOLOv4算法及其在行人检测中的应用 |
| |
作者姓名: | 宣勇 韩超 沙文瀚 |
| |
作者单位: | 1. 安徽工程大学高端装备先进感知与智能控制教育部重点实验室;2. 奇瑞新能源汽车股份有限公司 |
| |
摘 要: | 针对Tiny YOLOv4目标检测算法在行人检测中精确度低和召回率不高的问题,对特征提取网络及预测网络进行改进。在特征提取网络部分采用深度可分离卷积网络取代传统卷积网络,这减少了参数并降低了计算量;将注意力机制模块加入特征提取网络中以增强检测目标的感兴趣区域,提高检测精确度;在预测网络部分增加一个预测尺度,对增加的尺度进行特征增强处理,以提升目标检测的召回率。实验结果表明,与原算法相比,改进后的Tiny YOLOv4算法的检测精确度提升了7.1%,召回率提升了6.6%。
|
关 键 词: | 机器视觉 目标检测 深度可分离卷积 注意力机制 尺度增强预测 |
|