基于空间注意力和可变形卷积的无人机田间障碍物检测

作者：杜小强; 李卓林; 马锃宏; 杨振华; 王大帅

来源：农业机械学报, 2023, 54(02): 275-283.

DOI：10.6041/j.issn.1000-1298.2023.02.028

• 田间障碍物
• Mask R-CNN
• 空间注意力
• 可变形卷积

为了解决植保无人机作业时，传统田间障碍物识别方法依赖人工提取特征，计算耗时较长，难以实现在非结构化田间环境下实时作业识别的问题，提出一种优化的Mask R-CNN模型的非结构化农田障碍物实例分割方法。以ResNet-50残差网络为基础，将空间注意力(Spatial attention, SA)引入残差结构，聚焦跟踪目标的显著性表观特征并主动抑制噪声等无用特征的影响；引入可变形卷积(Deformable convolution, DCN),通过加入偏移量，增大感受野，提高模型的鲁棒性。构建包含农田典型障碍物的数据集，通过对比试验研究在ResNet残差网络结构中的不同阶段中加入空间注意力和可变形卷积时的模型性能差异。结果表明，与Mask R-CNN原型网络相比，在ResNet的阶段2、阶段3、阶段5加入空间注意力和可变形卷积后，改进Mask R-CNN的边界框(Bbox)和掩膜(Mask)的平均精度均值(mAP)分别从64.5%、56.9%提高到71.3%、62.3%。本文提出的改进Mask R-CNN可以很好地实现农田障碍物检测，可为植保无人机在非结构化农田环境下安全高效工作提供技术支撑。

• 单位
  中国科学院深圳先进技术研究院; 浙江理工大学; 龙泉市菇源自动化设备有限公司