摘要
本发明提供了一种基于目标显著性掩模的无监督螺丝图像分类与分割方法,属于图像处理领域,该方法包括:获取原始螺丝图像并转换为灰度图像,并基于无监督分割框架,提取显著性掩模,得到显著性图像;利用带有改进去噪锚框的检测变换器进行特征提取,得到测试特征和质心特征;根据测试特征和质心特征得到初步分割图,并再次利用显著性检测对分割图进行处理,得到分割图像,完成对螺丝部件的分割;将相同的测试图像通过缺陷检测模型,得到缺陷定位模型,并根据颜色得到缺陷部件的名称,完成对缺陷部件的分类。本发明解决了现有分类分割方法处理螺丝图像精度低、适合度差以及难以准确分割螺丝图像的问题。
技术关键词
测试特征
分割方法
螺丝部件
掩模
变换器
开源计算机视觉库
贪婪算法
无监督
K均值算法
训练集
像素
区分缺陷
图像块
标签
颜色
注意力机制
聚类