Ddpm Fake Icons

🚀 無条件DDPM拡散を用いたアイコン生成

このプロジェクトは、DDPM拡散モデルを用いて無条件のアイコン生成を行います。特定のデータセットを使って訓練され、独自のアイコンを生成することができます。

🚀 クイックスタート

データセット

訓練に使用されたデータセットは、Kaggle - Icons-50 です。

参考コード

訓練の参考コードは、Kaggle - Generating Fake Icons using DDPM Diffusion を参照しています。

画像サンプル

📦 インストール

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()

📄 ライセンス

このプロジェクトは、MITライセンスの下で公開されています。