摘要
近年来,高分遥感影像技术的快速发展为铁路沿线地物检测提供了一种重要技术手段。基于回归的一阶段目标检测方法YOLOv4具有检测精度高、速度快等优点,但用于遥感影像检测时仍然存在部分细节特征信息丢失导致的小目标漏检,以及进行大面积地物检测时效率低的问题。为此,提出改进YOLOv4网络模型对遥感影像铁路沿线地物进行检测。首先,设计由卷积、批量归一化和Mish激活函数组成的CBM(convolution batch normalization mish)模块,并采用DCBM(double CBM)模块作为密集连接网络(DenseNet)的传输层用于YOLOv4网络特征提取以实现地物特征传递和信息重用,增强小目标地物的检测能力,降低漏检率;然后针对YOLOv4在大面积检测时效率不高和模型参数空间较大的缺陷,将压缩激励SE(squeeze excitation)通道注意机制用于骨干网中跨阶段局部单元(cross stage partial,CSP)的每个残差单元之后,减少SE注意模块的重复调用次数,使其能够在提高网络性能的同时降低模型参数量从而提高检测效率;最后,针对长条形状的铁路目标提取困难问题,在网络结果输出之前引入改进的通道空间注意力机制ICBAM(improved convolutional block attention module)保留原始特征信息,解决铁路目标特征提取能力差的问题,提高铁路中大尺度目标的检测效率。为验证所提方法的有效性,选取2048张分辨率为1920×1080的某段铁路沿线遥感影像地物样本数据,将其中的铁路、房屋、楼宇建筑、农田和水池作为检测目标进行实验,并与当前流行的目标检测方法进行对比。结果表明,改进方法不仅增强了对小目标地物的检测能力,提高了地物检测精度和速度,而且提高了大尺度目标的检测效率。与YOLOv4算法相比,mAP提高了2.11%,准确率提高了2.93%,召回率提高了3.79%,模型大小减少了8.53%。所提方法为当前应用高速铁路沿线遥感影像地物快速精准检
In recent years,the rapid development of high-resolution remote sensing technology has provided an effective technical means for detecting ground objects along the railway.The regression-based one-stage target detection method YOLOv4 has the advantages of high detection accuracy and fast speed.However,when it is used for remote sensing image detection,small targets are still missed due to the loss of some detailed feature information,and large-area ground object detection.Due to the problem of low efficiency,this paper improves the YOLOv4 network model to detect ground objects along the railway in remote sensing images.This paper improves the YOLOv4 network model to detect the ground features in remote sensing images along the railway.First,the CBM(Convolution Batch Normalization Mish)module is designed with composing of convolution,batch normalization,and Mish activation,and the DCBM(Double CBM)module is used for the transmission layer of the densely connected network(DenseNet)for the YOLOv4 network feature extraction.It can achieve feature transfer and information reuse and enhance small target feature detection capabilities.Then,to address the defects of YOLOv4 in the inefficiency of large area detection and the large space of model parameters,the SE(Squeeze Excitation)channel attention mechanism is used after each residual cell of Cross Stage Partial(CSP)in the backbone network to reduce the number of repeated calls of the SE attention module.Hence the performance of the network is improved while reducing the number of model parameters and improving detection efficiency.Finally,for the problem of difficult extraction of railroad targets in images,an improved channel space attention mechanism ICBAM(Improved Convolutional Block Attention Module)is introduced before the network result output to retain the original feature information.It can solve the problem of poor feature extraction ability of railroad targets,and improve the detection efficiency of large-scale targets.To verify the effectiveness of the propos
作者
王阳萍
韩淑梅
杨景玉
党建武
张占平
WANG Yang-ping;HAN Shu-mei;YANG Jing-yu;DANG Jian-wu;ZHANG Zhan-ping(School of Electronic and Information Engineering,Lanzhou Jiaotong University,Lanzhou 730070,China;Gansu Engineering Research Center for Artificial Intelligence and Graphic&Image Processing,Lanzhou 730070,China)
出处
《光谱学与光谱分析》
SCIE
EI
CAS
CSCD
北大核心
2022年第10期3275-3282,共8页
Spectroscopy and Spectral Analysis
基金
国家自然科学基金项目(61763025,62067006)
甘肃省教育科技创新项目(2021jyjbgs-05)
甘肃省高校产业支撑计划项目(2020C-19)
国家市场监督管理总局科技计划项目(2019MK150)
兰州市科技计划项目(2019-4-49,2020-1-7)
天津大学-兰州交通大学自主创新基金合作项目
甘肃省高等学校青年博士基金项目(2021QB-053)资助。
关键词
地物检测
铁路沿线
遥感影像
YOLOv4
注意力机制
Ground target detection
Along with the railway
Remote sensing image
YOLOv4
Attention mechanism
