嵌入遮挡关系模块的SSD模型的输电线路图像金具检测

作者：赵振兵^*; 江爱雪; 戚银城; 张薇; 赵文清

来源：智能系统学报, 2020, 15(04): 656-662.

• 输电线路金具
• 遮挡度
• 遮挡关系描述
• 遮挡关系模块
• SSD
• 标注框
• 目标检测
• 深度学习

为了提升深度学习目标检测模型在输电线路金具自动化检测任务中的准确率,针对金具检测数据集中金具目标标注框之间不可避免地广泛存在相交而导致金具目标检测定位不准确的问题,本文利用相交区域的相似性作为金具目标的上下文信息,提出目标间遮挡关系的描述方法,用于规则性描述图像中金具目标间的相互遮挡,设计遮挡关系模块,并将其嵌入到单次多框检测器(single shot multibox detector, SSD)模型中。为了验证嵌入遮挡关系模块的SSD模型的有效性,选择了8类目标标注框普遍存在相交的小目标金具进行实验,实验使用的金具检测数据集的训练集和测试集中金具目标数分别为6 271和1 713。实验证明,原始SSD模型的平均精度均值(mean average precision, mAP)为72.10%,嵌入遮挡关系模块的SSD模型的m AP为76.56%,性能提升了4.46%。

• 单位
  电子工程学院; 华北电力大学