摘要
本发明公开了一种基于文本引导的弱监督图像缺陷分割方法及系统,包括:获取工业缺陷图像数据集;建立图像分割模型并训练,其中:初步预测模块用于提取多尺度图像块特征输入分割解码器获得初步分割预测结果;文本引导模块用于构建前景文本和背景文本并输入CLIP模型的文本编码器,生成前景文本特征集合和背景文本特征集合后拼接再与多尺度图像块特征输入上下文解码器;获得图文相关性得分图并与初步分割预测结果进行加权,获得最终分割预测结果;利用训练好的图像分割模型对待分割工业缺陷图像进行检测,获得最终分割预测结果。解决了工业检测任务中样本稀缺、标注成本高及模型泛化能力不足等问题,提升了图像缺陷检测的精度与效率。
技术关键词
图像缺陷分割方法
多尺度图像块特征
前景文本
图像分割模型
辅助分类器
解码器
像素点
卷积模块
标签
输出特征
文本编码器
工业
视觉特征
缺陷类别
图文