Defect detection of PCB based on improved YOLOv4 algorithm
CSTR:
Author:
Affiliation:

Faculty of Intelligent Manufacturing,Wuyi University,Jiangmen Guangdong 529020,China

Clc Number:

TP391;TP18

  • Article
  • | |
  • Metrics
  • | |
  • Related [20]
  • | | |
  • Comments
    Abstract:

    Reference template is used in most methods of PCB defect detection,which is very time consuming and causes a big position error. YOLOv4 is fast but it misses the object easily in PCB detection and its accuracy is not high in detecting the small object. Therefore, the method of PCB defect detection based on improved YOLOv4 algorithm is proposed. Firstly, CSPDarknet53 is used as backbone and the structure of single feature layer is adopted, which avoids the prior boxes assignment problem caused by data imbalance. Then, five convolutions are improved using CSP to increase further the ability offeature extract.Finally, prior boxes are gotten by using K-means++ to improve the training effect. In the experiment, Peking University PCB public dataset is used for training. The result shows that mean average precision of our algorithm achieves 98.71% and it has a better performance compared with other several classical object detection algorithms.

    Reference
    Cited by
    Comments
    Comments
    分享到微博
    Submit
Get Citation
Share
Article Metrics
  • Abstract:66
  • PDF: 287
  • HTML: 0
  • Cited by: 0
History
  • Online: August 09,2024
Article QR Code