首页|期刊导航|北京信息科技大学学报(自然科学版)|基于YOLO-CH的PCB焊点缺陷检测算法

基于YOLO-CH的PCB焊点缺陷检测算法OA

A PCB solder joint defect detection algorithm based on YOLO-CH

中文摘要英文摘要

为解决印制电路板(printed-circuit board,PCB)焊点缺陷检测领域存在的小目标特征易丢失、检测精度低以及模型参数量过多的问题,提出了一种基于改进YOLOv8算法的PCB焊点检测算法YOLO-CH.在YOLOv8特征金字塔与路径聚合网络基础上新增高分辨率P2检测层,将下采样倍数降至4倍,结合渐进式特征融合与双向特征交互策略,有效保留小目标的纹理、边缘等细节;设计了一种通道与边缘上下文聚合(channel and edge-wise context aggregation,CECA)双阶段注意力模块,采用通道注意力与坐标注意力级联的结构,增强小目标特征表示的同时保持模型轻量化;设计三维复合加权损失函数,缓解大小目标优化不平衡问题.基于自制PCB焊点缺陷数据集的实验结果表明,相较于YOLOv8算法,所提算法的mAP50提升了4.5百分点,mAP50-95提升了4.6百分点,召回率提升了5.3百分点,浮点运算量(floating-point operations,FLOPs)降低至7.09×1010,有效降低了模型复杂度,提升了缺陷检测效果.

To address the problems of easy loss of small-target features,low detection accuracy,and excessive model parameters in the field of printed-circuit board(PCB)solder joint defect detection,a PCB solder joint detection algorithm named YOLO-CH based on an improved YOLOv8 algorithm was proposed.Based on the feature pyramid and path aggregation network of YOLOv8,a high-resolution P2 detection layer was added,reducing the downsampling stride to four.Combined with progressive feature fusion and bidirectional feature interaction strategies,key details such as texture and edges of small targets were effectively preserved.A channel and edge-wise context aggregation(CECA)two-stage attention module was designed,which adopts a cascaded structure of channel attention and coordinate attention to enhance the feature representation of small targets while keeping the model lightweight.A three-dimensional composite weighted loss function was designed to alleviate the imbalance in optimization between large and small targets.Experimental results on a self-built PCB solder joint defect dataset show that compared with YOLOv8 algorithm,the proposed algorithm improves the mAP50 by 4.5 percentage points,mAP50-95 by 4.6 percentage points,and recall by 5.3 percentage points,while reducing the floating-point operations(FLOPs)to 7.09×1010,effectively lowering model complexity and improving defect detection performance.

孙雪怡;王一群;陈雯柏;耿悦;汪三元

北京信息科技大学 自动化学院,北京 100192北京信息科技大学 自动化学院,北京 100192北京信息科技大学 自动化学院,北京 100192北京信息科技大学 自动化学院,北京 100192北京信息科技大学 自动化学院,北京 100192

信息技术与安全科学

焊点缺陷检测小目标注意力机制损失函数

solder joint defect detectionsmall targetattention mechanismloss function

《北京信息科技大学学报(自然科学版)》 2026 (4)

42-51,10

国家自然科学基金重大研究计划项目(92267110)国家自然科学基金面上项目(62276028)北京市自然科学基金-小米联合基金重点研究专题(L233006)

10.16508/j.cnki.11-5866/n.2026.04.005

评论