A Random Forest Algorithm for PCB SMD Defect Detection

A PCB surface mounted devices (SMD) defect detection algorithm based on random forest was proposed to detect pin defects, resistor defects and capacitor defects. It can solve the problem of low detection accuracy and low efficiency caused by dense layout and small size of SMD. Firstly, the PCB image is preprocessed, which includes image graying, image stitching, geometric correction, component positioning and image denoising. Secondly, the SMD to be tested is divided into sub-regions. The shape features, gray features and texture features of the sub-regions are extracted. Then the random forest model is established and the CART decision tree is selected as the basic classifier. Each decision tree in the random forest makes a prediction according to the sample characteristics. Finally, all the decision tree predictions are voted to get the final prediction. The experimental results show that the average accuracy of this method for pin defect, resistor defect and capacitor defect detection is 97.7 %, 97.0 % and 96.7%. The detection time for a PCB board containing 200 SMDs is only 3.78 s.