公开(公告)号：CN115984236A

公开(公告)日：2023-04-18

申请号：CN202310057818.6

申请日：2023-01-17

Applicant: 重庆邮电大学

Inventor： 文德军 ,  周应华

IPC: G06T7/00 ,  G06V10/80 ,  G06V10/764 ,  G06V10/82 ,  G06N3/0464 ,  G06N3/048 ,  G06N3/08

Abstract: 本发明涉及一种基于改进YOLOv5n的计算机主板元器件检测方法，属于目标检测领域。该方法包括：S1：收集计算机主板图像并对元器件进行标注，得到数据集；S2：划分训练集、验证集和测试集；S3：构建改进YOLOv5n元器件检测模型：S31：构建主干特征提取网络、颈部特征融合网络和头部检测网络；S32：在主干特征提取网络中嵌入三个CA注意力机制；S4：使用构建的改进YOLOv5n元器件检测模型在训练集上训练，并在验证集上验证模型的检测能力，取验证集上表现最好的权重文件在测试集上评估结果；S5：将训练好的改进YOLOv5n元器件检测模型对待检测的计算机主板图像进行元器件的精准且快速的定位和分类。