一种基于扩散模型的图像生成方法

本发明提供了一种基于扩散模型的图像生成方法，涉及图像处理技术领域，该方法包括构建并初始化应用于图像生成领域的初始扩散模型，并对初始扩散模型进行二值化处理；将经二值化处理的扩散模型的所有二值结构替换为时间步友好的二值扩散模型结构，得到二值扩散模型；利用空间分块蒸馏对二值扩散模型进行训练，得到训练后的二值扩散模型；利用训练后的二值扩散模型生成图像任务。本发明解决了传统二值扩散模型生成图像时的崩溃问题。