西南交通大学学报2024，Vol.59Issue(6)：1294-1304,11.DOI:10.3969/j.issn.0258-2724.20220074
频谱池化与混洗注意力增强的铁路异物轻量检测
Lightweight Detection of Railway Object Intrusion Based on Spectral Pooling and Shuffled-Convolutional Block Attention Module Enhancement
摘要
Abstract
In infrared low-light scenes,railway object intrusion detection faces low detection accuracy,and it is difficult to achieve lightweight real-time detection.Therefore,a lightweight detection method of railway object intrusion based on convolutional block attention module(CBAM)enhancement was proposed.Firstly,the Darknet53 feature extraction network was improved by deep separable convolution to achieve lightweight extraction of railway object intrusion characteristics in infrared low-light scenes.Secondly,semantic-guided infrared spectral pooling was used for feature enhancement to improve the feature quality of infrared image downsampling.Then,a shuffled-CBAM was proposed to achieve feature extraction and fusion of key infrared targets and improve the accuracy of infrared target detection.Finally,the lightweight anchor-free network was used to predict the output result of railway object intrusion,overcoming the deficiency of poor real-time performance due to non-maximum value suppression operation with anchor frame detection,and it reduced calculation load and speeded up the detection efficiency.The experimental results show that the lightweight model has higher detection accuracy,and the size of the model is reduced by 179.01 MB after the improvement.The detection rate is increased to 39 frames/s,which is 3.9 times that of the YOLOv4 method.Compared with other detection methods,the proposed method can detect infrared railway object intrusion quickly and accurately.关键词
异物检测/红外弱光/混洗注意力/轻量化检测/高速铁路Key words
objection intrusion detection/infrared low-light scene/shuffled-CBAM/lightweight detection/high-speed railway分类
计算机与自动化引用本文复制引用
陈永,王镇,张娇娇..频谱池化与混洗注意力增强的铁路异物轻量检测[J].西南交通大学学报,2024,59(6):1294-1304,11.基金项目
国家自然科学基金项目(62462043,61963023) 兰州交通大学重点研发项目(ZDYF2304) （62462043,61963023）
兰州交通大学天佑创新团队(TY202003). （TY202003）
