许可证: mit 语言:

• 英文 任务标签: 无条件图像生成

使用DDPM扩散模型的无条件图标生成

训练所用数据集: Kaggle - Icons-50

训练参考代码: Kaggle - 使用DDPM扩散生成虚拟图标

图像样本

Google Colab使用示例代码

安装必要包

!python3 -m pip install diffusers==0.21.* accelerate

使用DDPM管道生成图像

from matplotlib import pyplot as plt
from diffusers import DDPMPipeline

ddpm = DDPMPipeline.from_pretrained("mayur7garg/ddpm-fake-icons")
ddpm.to("cuda")

inference_image = ddpm(
    batch_size = 36,
    num_inference_steps = 1000
).images

plt.figure(figsize = (18, 18), dpi = 120)

for i, image in enumerate(inference_image):
    plt.subplot(6, 6, i + 1)
    plt.imshow(image)
    plt.axis(False)
plt.tight_layout()
plt.show()