Journal of Applied Optics, Volume. 44, Issue 3, 621(2023)

Surface defect detection of patch diode based on improved YOLO-V4

Liequan WU1, Zhifeng ZHOU1、*, Zhiling ZHU1, Wei ZHANG2, and Yong WANG3
Author Affiliations
  • 1School of Mechanical and Automotive Engineering, Shanghai University of Engineering Science, Shanghai 201620, China
  • 2Shanghai Compass Satellite Navigation Technology Co.,Ltd., Shanghai 201801, China
  • 3State Grid Siji Location Service Co.,Ltd., Beijing 102211, China
  • show less

    Aiming at low efficiency of traditional visual detection method and shallow model as well as low semantic character of target detection algorithm based on manual features, an surface defect detection method of patch diode based on improved YOLO-V4 was proposed. Firstly, DenseNet was used in CSP1 module to replace ResNet in original network, considering that gradient disappeared with network deepening and feature redundancy as well as parameters were reduced. Then, to realize cross-dimensional interaction of feature information and make the network pay more attention to important information, the three-branch attention mechanism module was introduced after CSP1 module, and features were fused with FPN+PANet. CBL×5 module was replaced by CSP2, which reduced computation of network and improved detection speed of algorithm. Finally, the Focal Loss function was optimized and weight was added to positive and negative samples to solve the imbalance problems. The detection precision (P), recall ratio (R) and mean average precision (mAP) of the algorithm are 2.98%, 2.65% and 2.92% higher than that of YOLO-V4, respectively, which shows that the improved YOLO-V4 can effectively detect the surface defects of patch diode.

    Tools

    Get Citation

    Copy Citation Text

    Liequan WU, Zhifeng ZHOU, Zhiling ZHU, Wei ZHANG, Yong WANG. Surface defect detection of patch diode based on improved YOLO-V4[J]. Journal of Applied Optics, 2023, 44(3): 621

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Research Articles

    Received: Jun. 13, 2022

    Accepted: --

    Published Online: Jun. 19, 2023

    The Author Email: Zhifeng ZHOU (zhousjtu@126.com)

    DOI:10.5768/JAO202344.0303007

    Topics