基于改进YOLOv4的太阳能电池板缺陷检测

Solar Panel Defect Detection Based on Improved YOLOv4

  • 摘要: 针对太阳能电池板缺陷检测精度低、检测速度慢、模型体积大的问题,提出一种改进YOLOv4的检测模型。首先,采用GhostNet替换YOLOv4中CSPDarknet-53实现模型轻量化;其次,在模型结构中引入深度可分离卷积进一步减少模型参数,提升模型快速性;再次,在模型中引入改进高效通道注意力(ECA)机制,提高检测精度;最后,使用S-T-ReLU激活函数替换原YOLOv4中ReLU激活函数,进一步提高检测精度。结果表明:改进模型的检测效果更佳,相比较原始模型,mAP提高1.06%,每秒浮点运算次数(FLOPs)降低89.11%,模型体积减小82.61%,模型参数量减少82.77%,每秒帧数(FPS)提升35.34%,证明了所提算法的有效性。

     

    Abstract: An improved YOLOv4 detection model is proposed to address the issues of low accuracy, slow detection speed and large model volume in defect detection of solar panels. Firstly, GhostNet is used to replace CSPLocknet-53 in YOLOv4 to achieve model lightweight. Secondly, introducing Depthwise Separable Convolution convolution into the model structure further reduces model parameters and improves model speed. Thirdly, an improved Efficient Channel Attention(ECA) mechanism is introduced into the model to improve detection accuracy. Finally, the S-T-ReLU activation function is used to replace the ReLU activation function in the original YOOv4 to further improve the detection accuracy. The results showed that the improved model had better detection performance. Compared with the original model, the mAP increased by 1.06%, the Floating Point Operation Per Second (FLOPs) decreased by 89.11%, the model volume decreased by 82.61%, the model parameter quantity decreased by 82.77%, and the Frames Per Second (FPS) increased by 35.34%, proving the effectiveness of the proposed algorithm.

     

/

返回文章
返回