计算机应用与软件2024,Vol.41Issue(9) :166-174.DOI:10.3969/j.issn.1000-386x.2024.09.024

基于改进卷积神经网络的小目标检测算法

SMALL TARGET DETECTION ALGORITHM BASED ON IMPROVED CONVOLUTIONAL NEURAL NETWORK

张明 余志强
计算机应用与软件2024,Vol.41Issue(9) :166-174.DOI:10.3969/j.issn.1000-386x.2024.09.024

基于改进卷积神经网络的小目标检测算法

SMALL TARGET DETECTION ALGORITHM BASED ON IMPROVED CONVOLUTIONAL NEURAL NETWORK

张明 1余志强2
扫码查看

作者信息

  • 1. 北京全路通信信号研究设计院集团有限公司 北京 100070;石家庄铁道大学电气与电子工程学院 河北石家庄 050043
  • 2. 北京全路通信信号研究设计院集团有限公司 北京 100070;石家庄铁道大学电气与电子工程学院 河北石家庄 050043;石家庄铁道大学河北省电磁环境效应与信息处理重点实验室 河北石家庄 050043
  • 折叠

摘要

对于在检测时存在小尺度检测目标漏检、不精确等问题,提出一种改进的YOLO v4算法模型(F-YOLO v4).利用改进的K均值聚类算法对数据集进行聚类,使得锚点框的大小更适用于目标检测;采用深度卷积和逐点卷积相结合的方法对通道内和通道间的卷积进行分离,从而改善了原有的残差块;采用通道注意力机制对骨干网络进行改进的同时在PANet网络中添加RFB模块,增强特征提取能力,从而提高了对小目标的检测效果.实验结果表明,F-YOLO v4算法在KITTI数据集上平均精度均值达到了93.67%,与原算法对比提高了1.52百分点,并且比较目前其他主流网络有着较高的精确度.

Abstract

In order to solve the problems of small-scale target missing detection and imprecision in detection,an improved YOLO v4 algorithm model(F-YOLO v4)is proposed.The improved K-means clustering algorithm was used to cluster the data set,which made the size of anchor frame more suitable for target detection.The convolution within and between channels was separated by the combination of deep convolution and point by point convolution,so as to improve the original residual block.The channel attention mechanism was used to improve the backbone network,and the RFB module was added to the PANet network to increase the number of residual blocks.The feature extraction ability was enhanced,and the detection effect of small target was improved.The experimental results show that the average accuracy of F-YOLO v4 algorithm on KITTI data set reaches 93.67%,which is 1.52 percentage points higher than the original algorithm,and has higher accuracy than other mainstream networks.

关键词

改进的YOLO/v4算法/深度可分离卷积/注意力机制/RFB模块/小目标检测

Key words

Improved YOLO v4 algorithm/Depth separable convolution/Attention mechanism/RFB module/Small target detection

引用本文复制引用

基金项目

国家自然科学基金项目(11872257)

河北省自然科学基金面上项目(E2018210144)

河北省重点研发计划项目(20354501D)

石家庄市军民融合项目(201060104A)

国家轨道交通电磁环境效应研究与测试平台建设项目(50200011800)

出版年

2024
计算机应用与软件
上海市计算技术研究所 上海计算机软件技术开发中心

计算机应用与软件

CSTPCD北大核心
影响因子:0.615
ISSN:1000-386X
段落导航相关论文