一种基于图网络的PCB板元器件图像目标检测方法技术

本发明专利技术公开一种基于图网络的PCB板元器件图像目标检测方法，包括步骤：将PCB图片样本数据集中五分之四的样本分为训练集，其余样本分为验证集，利用生成式对抗网络对训练集的所有样本学习生成新样本；每个新样本利用超像素构建一个K最近邻接矩阵；建立RPN网络，提取多个PCB超像素训练集样本图片的边框位置并通过损失函数不断修正靠近正确的标注数据框的概率；采用图卷积网络对特征图进行特性细化生成特征图信息；利用相似度预测算法对前一个阶段的每个特征图进行分类，再计算每个图对元器件模板的相似性概率并基于概率归类为相应元器件模板的类别。本发明专利技术实现快速有效的分类标注出PCB板上的各个元器件，极大地节省了人力成本。极大地节省了人力成本。极大地节省了人力成本。

【技术实现步骤摘要】
一种基于图网络的PCB板元器件图像目标检测方法


[0001]本专利技术涉及图像处理
，尤其涉及一种基于图网络的PCB板元器件图像目标检测方法。

技术介绍

[0002]PCB板元器件的检测是自动化检测硬件的一个必要步骤，传统意义上的检测由质检工程师进行目检手动框定元器件(例如：电阻、电容)，然后识别组件类型，一片PCB板一般包含上百个元器件，质检工程师需要了解各样式的电子元器件才能标注准确，质检工程师需要花费大约两天时间标注一片PCB板，需要耗费大量的时间与精力，门槛高，标注成本昂贵；因此，有必要提供一种快速标注与识别元器件的方法，借助于深度学习中条件生成式对抗网络消除图像模糊性，与超像素、图神经网络的思想构建目标检测模型框架，能够快速有效的分类标注出PCB板上的各个元器件，节省大量人力成本。

技术实现思路

[0003]本专利技术的目的是克服现有技术的不足，提供一种基于图网络的PCB板元器件图像目标检测方法。
[0004]本专利技术的技术方案如下：
[0005]一种基于图网络的PCB板元器件图像目标检测方法，本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于图网络的PCB板元器件图像目标检测方法，其特征在于，包括如下步骤：S1，获取PCB图片样本数据集，并将该PCB图片样本数据集中五分之四的样本划分为训练集，另外五分之一的样本划分为验证集；利用GANs生成式对抗网络对训练集的所有样本学习生成新样本；S2，对于步骤S1生成的每一个新样本利用超像素构建一个K最近邻接矩阵，权值为：；式中xi和xj分别是超像素i和j的二维坐标，σx是放缩参数；S3，建立RPN网络，提取多个PCB超像素训练集样本图片的边框位置，并通过函数不断平移缩放修正靠近正确的标注数据框的概率，从而为下一步骤提供特征图；S4，采用图卷积网络并结合元器件模板图对步骤S3提供的特征图进行特性细化，生成特征图信息；S5，针对步骤S4生成的特征图信息利用相似度预测算法对步骤S3产生的每个特征图进行分类；再利用相似度预测算法计算每个被步骤S3和步骤S4处理后的图对元器件模板的相似性概率，并基于概率归类为相应元器件模板的类别。2.根据权利要求1所述的基于图网络的PCB板元器件图像目标检测方法，其特征在于，所述步骤S4进行特性细化的具体实现过程如下：令图卷积网络中的图的实体表示成节点，节点间的连接表示成边，系统层面的特征用全局属性表示，节点表示为v...

【专利技术属性】
技术研发人员：冯建设，花霖，张建宇，
申请(专利权)人：深圳市信润富联数字科技有限公司，
类型：发明
国别省市：