首页|基于知识蒸馏的PCB板缺陷检测方法

基于知识蒸馏的PCB板缺陷检测方法

扫码查看
为了提升PCB板表面缺陷检测的效率,提出了一种基于知识蒸馏的PCB板缺陷检测方法.搭建了一大一小两个网络模型,大模型检测精度高,检测效率低;小模型检测效率高,但检测精度低.使用知识蒸馏方法将大模型的"知识"蒸馏至小模型中,蒸馏不改变小模型的检测效率,但会使小模型的检测精度获得巨大提升.试验结果表明:大模型在交并比为0.5时的平均精度高达0.938,经知识蒸馏,模型的检测效率在RTX4090上帧率可达454.5.所提方法成功实现了对印刷电路板缺陷的高精度高效检测,具有重要意义.
PCB Board Defect Detection Method Based on Knowledge Distillation
In order to enhance the efficiency of surface defect detection on PCB boards,a PCB board defect detection method based on knowledge distillation was proposed.Two network models were built:a large one with high accuracy but low efficiency,and a small one with high efficiency but lower accuracy.Using knowledge distillation method to distill the"knowledge"of the large model into the small model did not change the detection efficiency of the small model,but it would greatly improve the detection accuracy of the small model.The experimental results show that the average accuracy of the large model is as high as 0.938 while the intersection to union ratio is 0.5.After knowledge distillation.The detection efficiency of the small model on RTX4090 can reach a frame rate of 454.5.The proposed method has successfully achieved high-precision and efficient detection of defects in PCBs,which is of great significance.

PCB detectiondefect detectionmodel compressionknowledge distillation

陈政宇、张峰、张士文

展开 >

上海交通大学电子信息与电气工程学院,上海 200240

PCB板检测 缺陷检测 模型压缩 知识蒸馏

2024

电气自动化
上海电气自动化设计研究所有限公司 上海市自动化学会

电气自动化

CSTPCD
影响因子:0.377
ISSN:1000-3886
年,卷(期):2024.46(6)