图像局部编辑方法、装置、设备及介质

本发明涉及人工智能领域，可应用于金融、医疗领域，本发明公开了图像局部编辑方法、装置、设备及介质，该方法包括获取原始图像、图像待替换位置掩码和提示文本，并获取提示文本对应的词元嵌入；其中，图像待替换位置掩码用于对原始图像的对应位置进行图像编辑；获取与第一扩散模型连接的第二扩散模型所输出的注意力统计向量；将提示文本对应的词元嵌入、注意力统计向量、原始图像及图像待替换位置掩码输入至第一扩散模型，得到与原始图像对应且采用了图像待替换位置掩码编辑的编辑后图像。本发明通过引入注意力平滑结构和知识蒸馏机制，提升了扩散模型在局部图像编辑任务中的一致性与可控性。