淄博市知识产权公共服务平台

导航
专利号 2023117814152 申请日 2023-12-21 专利名称 基于反馈强化学习的小样本表面缺陷图像生成方法及系统
授权日 2024-11-15 专利权人 山东大学 发明人 宋然;房体育;屈益民;张伟;魏致远
主分类号 G06T7/00 关键词 应用领域
摘要 本发明公开了一种基于反馈强化学习的小样本表面缺陷图像生成方法及系统,该方法包括:获取少量表面缺陷图像样本,进行语义标签标注,构建训练集;构建条件扩散模型,利用训练集训练条件扩散模型,完成条件扩散模型的初步微调;扩充语义标签,利用微调后的条件扩散模型生成每一语义标签下的多张初始表面缺陷图像;对每一语义标签下的多张初始缺陷图像进行人工评分,进而构建评价数据集;构建奖励模型,利用评价数据集训练奖励模型;扩充语义标签,构建语义标签库,并基于奖励模型二次优化条件扩散模型,生成每一语义标签所对应的最优表面缺陷图像。本发明能够有效提高表面缺陷图像样本生成的性能和数据增强的能力。
创新点
技术分类 标 签 战兴产业 相关服务业    新技术与创新创业服务
运营方式 合作方式
联系人 联系电话 电子邮箱
详细说明
【关 闭】