针对电力设备检测中小目标识别精度低的问题,提出一种基于RetinaNet的电力设备目标检测方法.通过K-means聚类方法优化原始网络的锚点框尺寸.然后在特征融合中加入拥有更高分辨率的浅层特征图,解决通过多层卷积后特征图包含信息过少的问题.在此基础上,引入ECA(Efficient Channel Attention)注意力机制,使网络定位电力设备的有效特征,抑制无用特征信息.实验结果表明,相比原始方法,本文方法对于电塔、销钉、工程车、绝缘子、电杆5种电力设备的平均识别精度提升了18.1个百分点,表明改进后的方法能显著提高电力设备的检测水平.
A RetinaNet-based target detection method for power equipment detection is proposed for the problem of low accuracy of small target recognition in power equipment detection.The anchor box size of original network is optimized by K-means clus-tering method firstly.Then shallow feature maps with higher resolution are added to feature fusion to solve the problem that the feature maps contain too little information after convolution through multiple layers.Based on this,ECA(Efficient Channel At-tention)attention mechanism is introduced to enable the network to locate the effective features of power devices and suppress the useless feature information.The experimental results show that compared with the original method,the average recognition accuracy of the method in the paper is improved by 18.1 percentage points for five types of power equipment:electric towers,pins,construction vehicles,insulators and poles,which indicates that the improved method can significantly improve the detec-tion level of power equipment.