基于扩散模型的图像增强方法、装置、设备及存储介质
摘要:
本发明公开基于扩散模型的图像增强方法、装置、设备及存储介质,方法包括:获取待增强的目标图像与图像增强指令,编码得到编码特征图和文本编码;将编码特征图和文本编码输入到预先训练好的目标图像增强网络中;按照预设噪声添加规则和预设步数,逐步向编码特征图中添加高斯噪声,得到服从高斯分布的目标噪声图像,并确定每一步添加高斯噪声后的结果图像中的预测噪声;基于交叉注意力机制,对目标噪声图像中与文本编码对应的区域进行图像增强,得到加噪增强图像;按照预设噪声去除规则和预设步数,逐步从加噪增强图像中去除每一步的预测噪声,得到去噪后图像;对去噪后图像进行解码,得到增强后图像。本发明有效地提高了对特征缺失较多的图像的增强效果。
0/0