- 专利标题: 图像修复模型训练方法、装置、电子设备及存储介质
-
申请号: CN202311331636.X申请日: 2023-10-16
-
公开(公告)号: CN117079083B公开(公告)日: 2024-02-09
- 发明人: 温东超 , 梁玲燕 , 史宏志 , 赵雅倩 , 葛沅 , 崔星辰 , 张英杰
- 申请人: 苏州元脑智能科技有限公司
- 申请人地址: 江苏省苏州市吴中经济开发区郭巷街道官浦路1号9幢
- 专利权人: 苏州元脑智能科技有限公司
- 当前专利权人: 苏州元脑智能科技有限公司
- 当前专利权人地址: 江苏省苏州市吴中经济开发区郭巷街道官浦路1号9幢
- 代理机构: 北京润泽恒知识产权代理有限公司
- 代理商 黄佳
- 主分类号: G06V10/774
- IPC分类号: G06V10/774 ; G06V10/764 ; G06V10/82 ; G06T5/77 ; G06T5/60 ; G06N3/045 ; G06N3/0499 ; G06N3/08
摘要:
本申请提供了一种图像修复模型训练方法、装置、电子设备及存储介质。所述包括:对待处理图像进行切片处理,构建得到模型训练样本;基于待训练图像修复模型对所述模型训练样本进行处理,以得到所述模型训练样本的像素特征、结构特征和视觉感知特征;基于所述像素特征、所述结构特征和所述视觉感知特征,分别计算得到重构损失函数、结构相似损失函数和视觉感知损失函数;基于所述重构损失函数、所述结构相似损失函数和所述视觉感知损失函数,计算得到所述待训练图像修复模型的损失函数;在基于损失函数确定待训练图像修复模型收敛的情况下,得到图像修复模型。本申请可以有效控制重建区域的目标结构和视觉纹理,使得修复的图像具有更好的视觉效果。
公开/授权文献
- CN117079083A 图像修复模型训练方法、装置、电子设备及存储介质 公开/授权日:2023-11-17