首页|基于改进Cascade R-CNN的安全帽检测算法

基于改进Cascade R-CNN的安全帽检测算法

扫码查看
原文链接
  • 万方数据
  • 维普
针对安全帽检测中,目标形状、尺度变化大,易出现漏检、误检等问题,提出了一种基于改进级联基于区域的卷积神经网络(CascadeR-CNN)的安全帽检测算法.首先,对ResNet50进行改进形成D-ResNet50,利用可变形卷积仅增加少量参数就可增大感受野的特性,对特征提取网络的C2～C5卷积层进行重塑,提高网络对目标几何变换的适应能力和特征提取能力.其次,将D-ResNet50作为主干网络引入Cascade R-CNN,形成级联目标检测器,在每个阶段对正负样本重采样,抑制误检问题.再次,对递归特征金字塔进行改进,更高效地进行多尺度特征融合,并且基于反馈信息对特征进行二次处理,增强特征表达,提高网络的分类和定位能力.最后,使用Soft-非极大值抑制(Soft-NMS)进行后处理,进一步解决漏检问题.提出的方法在Hard hat workers数据集上的AP值相比检测基线提高了3.5％,与 SparseR-CNN、TridentNet、VFnet 等先进算法相比分别提升了4.7％、5.9％、2.3％等.
Safety helmet detection algorithm based on the improved Cascade R-CNN
Aiming at the problems of large changes in target shape and scale,which are prone to missed detection and false detection in safety helmet detection,a safety helmet detection algorithm based on improved Cascade Region-based Convolutional Neural Networks(Cascade R-CNN)is proposed.Firstly,ResNet50 is improved to form D-ResNet50,which can increase the perceptual field by using the feature of deformable convolution with only a small increase of parameters,and reshape the C2 to C5 convolutional layers of the feature extraction network to improve the network's adaptability to target geometric transformation and feature extraction capability.Secondly,D-ResNet50 is introduced into Cascade R-CNN as the backbone network to form a cascade target detector.Convolutional Neural Networks(CNN)to form a cascade target detector,resampling positive and negative samples at each stage to suppress the false detection problem.Thirdly,the recursive feature pyramid is improved to perform multi-scale feature fusion more efficiently,and the features are processed twice based on feedback information to enhance feature representation and improve the classification and localization ability of the network.Finally,post-processing is performed using Soft-Non-Maximum Suppression(Soft-NMS)to further solve the problem of missed detection.The proposed method improves the AP value on Hard hat workers dataset by 3.5％compared to the detection baseline,and by 4.7％,5.9％,2.3％,etc.compared to the advanced algorithms such as Sparse R-CNN,TridentNet,and VFnet,respectively.

safety helmet detectionmultiscale feature fusionfeedback connectiondeformable convolutionCascade R-CNNCARAFE

冯佩云、钱育蓉、范迎迎、魏宏杨、秦雨刚、莫王昊

展开 >

新疆大学软件学院,新疆乌鲁木齐 830091

新疆维吾尔自治区信号检测与处理重点实验室,新疆乌鲁木齐 830046

新疆大学软件工程重点实验室,新疆乌鲁木齐 830091

新疆财经大学信息管理学院,新疆乌鲁木齐 830012

展开 >

安全帽检测 多尺度特征融合 反馈连接 可变形卷积 Cascade R-CNN CARAFE

国家自然科学基金自治区科技厅国际合作项目国家自然科学基金联合基金重点项目自治区自然科学基金自治区科技计划青年科学基金

619660352020E01023U18032612021D01C0832022D01C83

2024

10.19304/J.ISSN1000-7180.2022.0838

微电子学与计算机
中国航天科技集团公司第九研究院第七七一研究所

微电子学与计算机

CSTPCD
影响因子：0.431
ISSN：1000-7180
年,卷(期)：2024.41(1)
  • 32