← 返回列表
基于稳定扩散模型的缺陷图像样本生成方法和装置
摘要文本
专利查询网 本申请公开了一种基于稳定扩散模型的缺陷图像样本生成方法和装置，涉及机器视觉领域，该方法根据输入预训练的稳定扩散模型的目标缺陷类型，从目标缺陷类型对应的一组缺陷特征映射图中随机选择目标缺陷特征映射图，基于随机生成的缺陷位置矩阵在任一非缺陷图像样本上随机生成缺陷区域，将目标缺陷特征映射图填充至缺陷区域，基于稳定扩散模型向任一非缺陷图像样本上的缺陷区域添加随机高斯噪声来进行稳定扩散，从而生成包含目标缺陷类型的缺陷特征的新缺陷图像样本。本申请生成新缺陷图像样本无需额外的缺陷标注，增强了缺陷图像样本的缺陷区域与背景图像之间的特征连续性，降低了对输入图像样本的采样计算量。
申请人信息
- 申请人:东声(苏州)智能科技有限公司
- 申请人地址:215000 江苏省苏州市工业园区酝慧路168号星洲大厦101、102、203、205、213室
- 发明人: 东声(苏州)智能科技有限公司
专利详细信息
| 项目 | 内容 |
|---|---|
| 专利名称 | 基于稳定扩散模型的缺陷图像样本生成方法和装置 |
| 专利类型 | 发明申请 |
| 申请号 | CN202311630718.4 |
| 申请日 | 2023/12/1 |
| 公告号 | CN117333740A |
| 公开日 | 2024/1/2 |
| IPC主分类号 | G06V10/774 |
| 权利人 | 东声(苏州)智能科技有限公司 |
| 发明人 | 韩旭; 颜聪; 陈彦龙 |
| 地址 | 江苏省苏州市工业园区酝慧路168号星洲大厦101、102、203、205、213室 |
专利主权项内容
1.一种基于稳定扩散模型的缺陷图像样本生成方法，其特征在于，包括：根据原始缺陷图像样本集的缺陷标注数据获取每个缺陷类型分别对应的一组缺陷特征映射图；根据输入预训练的稳定扩散模型的目标缺陷类型，从所述每个缺陷类型分别对应的一组缺陷特征映射图获得所述目标缺陷类型对应的一组缺陷特征映射图，并从所述目标缺陷类型对应的一组缺陷特征映射图中随机选择目标缺陷特征映射图，基于随机生成的缺陷位置矩阵在非缺陷图像样本集中的任一非缺陷图像样本上随机生成缺陷区域，将所述目标缺陷特征映射图填充至所述缺陷区域；基于所述稳定扩散模型，通过向所述任一非缺陷图像样本上的所述缺陷区域添加随机高斯噪声进行稳定扩散，生成包含所述目标缺陷类型的缺陷特征的新缺陷图像样本。 来自马-克-数-据