改进YOLOv5的非静止风机朝向检测混合算法
HYBRID ALGORITHM FOR NON-STATIONARY FAN ORIENTATION DETECTION BASED ON IMPROVED YOLOV5
-
摘要: 在计算机视觉广泛应用于各种工业的背景下,为了实现风力发电场的无人机智能化巡检,提出基于改进YOLOv5s网络模型的混合检测算法:引入α-CIoU改进原有交并比(IoU)损失函数,提高检测物体预测框的回归精度;在原网络中加入全局注意力模块,提高网络对目标位置和通道特征的全局提取能力;在检测层后结合FAST特征提取算法,提取风机叶片不同朝向的几何特征进行细粒化目标检测,提高整体算法的检测精度,降低误检率。实验结果表明该混合算法平均精度(mAP@0.5∶0.95)可达84.01%,较原算法提高9.56百分点,但每秒检测帧数(FPS)只下降了7帧/s。Abstract: In the context of computer vision being widely used in various industries, in order to realize the intelligent inspection in wind farm using unmanned aerial vehicle, a hybrid detection algorithm based on improved YOLOv5 is proposed. The α-CIoU was introduced by replacing the original IoU loss function to improve the precision of detection of high IoU object. The global attention module was added to the original network to enhance the global extraction ability of the network for target position and channel features. Features from accelerated segment test (FAST) algorithm were combined after the detection layer to capture the geometric features of different orientations of the fan blade to assist in fine-grained target detection, improving the detection accuracy of the algorithm and reducing the false detection rate. The experimental results show that the mean precision (mAP@0.5∶0.95) of the hybrid algorithm can reach 84.01%, which is 9.56 percentage points higher than the original algorithm, but the detection speed is only reduced by 7 FPS.
下载: