发明公开
- 专利标题: 基于扩散模型的图像增强方法、装置、设备及存储介质
-
申请号: CN202310922672.7申请日: 2023-07-26
-
公开(公告)号: CN116664450A公开(公告)日: 2023-08-29
- 发明人: 王红凯 , 徐昱 , 毛冬 , 戴波 , 陈祖歌 , 黄建平 , 李钟煦 , 郑怡 , 饶涵宇 , 李高磊
- 申请人: 国网浙江省电力有限公司信息通信分公司 , 国网信息通信产业集团有限公司 , 国网浙江省电力有限公司磐安县供电公司
- 申请人地址: 浙江省杭州市黄龙路8号641室; ;
- 专利权人: 国网浙江省电力有限公司信息通信分公司,国网信息通信产业集团有限公司,国网浙江省电力有限公司磐安县供电公司
- 当前专利权人: 国网浙江省电力有限公司信息通信分公司,国网信息通信产业集团有限公司,国网浙江省电力有限公司磐安县供电公司
- 当前专利权人地址: 浙江省杭州市黄龙路8号641室; ;
- 代理机构: 广州三环专利商标代理有限公司
- 代理商 薛梦
- 主分类号: G06T5/00
- IPC分类号: G06T5/00 ; G06V10/44 ; G06T9/00
摘要:
本发明公开基于扩散模型的图像增强方法、装置、设备及存储介质,方法包括:获取待增强的目标图像与图像增强指令,编码得到编码特征图和文本编码;将编码特征图和文本编码输入到预先训练好的目标图像增强网络中;按照预设噪声添加规则和预设步数,逐步向编码特征图中添加高斯噪声,得到服从高斯分布的目标噪声图像,并确定每一步添加高斯噪声后的结果图像中的预测噪声;基于交叉注意力机制,对目标噪声图像中与文本编码对应的区域进行图像增强,得到加噪增强图像;按照预设噪声去除规则和预设步数,逐步从加噪增强图像中去除每一步的预测噪声,得到去噪后图像;对去噪后图像进行解码,得到增强后图像。本发明有效地提高了对特征缺失较多的图像的增强效果。