摘要
本发明公开的一种小样本柔性IC基板的表面缺陷定位和分类方法,包括以下步骤:构造初始训练集,将具有特定外形特征的类别分别进行合并分类并进行标签增加;对图像进行预处理;训练缺陷目标检测模型;分离出圆孔类、划痕类和不规则轮廓类图像组成新子训练集;将新子训练集图像样本按标签细分成具体类别;抽取图像的HOG特征描述子,并构造数据样本,对圆孔类、划痕类和不规则轮廓类分别训练支持向量机模型,实现多分类;定位子训练集缺陷图像中缺陷位置;计算两种定位方式下子训练集图像缺陷定位结果与真实标签框的偏移量,获得缺陷位置坐标的最终预测结果,完成检测过程。本发明能够快速、高效、准确地实现小样本工业图像缺陷检测。
-
单位华南理工大学; 广州现代产业技术研究院