基于注意力与特征融合的工程机械目标检测方法

Object Detection Method of Construction Machinery Based on Attention and Feature Fusion

目的针对施工环境中工程机械目标大小不一、相互遮挡、工作形态各异等问题,提出一种基于注意力与特征融合的目标检测方法(AT–FFRCNN)。方法在主干网络中采用ResNet50和特征路径聚合网络PFPN,融合不同尺度的特征信息,在区域建议网络(RPN)和全连接层引入注意力机制,提高目标识别的能力,在损失函数中使用广义交并比(GIoU),提高目标框的准确性。结果实验表明,文中提出方法检测准确率比其他方法有较大提高,检测平均准确率(mAP)达到90%以上。结论能够较好地完成工程机械目标的检测任务。

The work aims to propose an object detection method based on attention and feature fusion(AT-FFRCNN)aiming at the problems of different size,mutual occlusion and different working forms of construction machinery objects in the construction environment.ResNet50 and feature path aggregation network PFPN were used in the backbone net-work to fuse feature information of different scales,and an attention mechanism was introduced into the region proposal network(RPN)and fully connected layer to improve the ability of target recognition,and generalized intersection over union(GIoU)was used in the loss function to improve the accuracy of the object box.Experiments indicated that the de-tection accuracy of the proposed method was greatly improved compared with other methods,and the average detection accuracy(mAP)reached more than 90%.The proposed method can complete the detection task of the construction ma-chinery better.