计算机应用

作者：蒋占军, 吴佰靖, 马龙, 廉敬

单位：兰州交通大学 电子与信息工程学院，兰州 730070

关键词：目标检测,水漂垃圾,Faster-RCNN,空洞卷积,多尺度特征融合,极化自注意力

基金：国家自然科学基金资助项目(62061023);甘肃省水利科学研究院基金资助项目(LZJT523029)

针对小目标水漂垃圾形态多变、分辨率低且信息有限，导致检测效果不理想的问题，提出一种改进的Faster-RCNN（Faster Regions with Convolutional Neural Network）水漂垃圾检测算法MP-Faster-RCNN（Faster-RCNN with Multi-scale feature and Polarized self-attention）。首先，建立黄河兰州段小目标水漂垃圾数据集，将空洞卷积结合ResNet-50代替原来的VGG-16（Visual Geometry Group 16）作为主干特征提取网络，扩大感受野以提取更多小目标特征；其次，在区域生成网络（RPN）利用多尺度特征，设置3×3和1×1的两层卷积，补偿单一滑动窗口造成的特征丢失；最后，在RPN前加入极化自注意力，进一步利用多尺度和通道特征提取更细粒度的多尺度空间信息和通道间依赖关系，生成具有全局特征的特征图，实现更精确的目标框定位。实验结果表明，MP-Faster-RCNN能有效提高水漂垃圾检测精度，与原始Faster-RCNN相比，平均精度均值（mAP）提高了6.37个百分点，模型大小从521 MB降到了108 MB，且在同一训练批次下收敛更快。

来源：2024年第3期

《计算机应用》期刊编辑部

查看计算机应用杂志2024年第3期