kyonimouto/hoyu-ai

Diffusion GANというコードを使ってつくりました

https://github.com/Zhendong-Wang/Diffusion-GAN

つかいかた 試してないので動かなかったらごめんなさい

• 環境をととのえる
  • 最近のNVIDIA製GPUがついたパソコンにLinuxを入れることをおすすめします
  • PytorchをCUDAありでインストールしてください
    • https://pytorch.org/get-started/locally/
    • conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia
  • Gitもインストールしてください
    • sudo apt install git
• Diffusion-GANをgithubからローカルにcloneしてください
  • git clone https://github.com/Zhendong-Wang/Diffusion-GAN
• diffusion-projected-ganというフォルダを開いてください
• ここの"Files and versions"から"best_model.pkl"をクリックしてダウンロードし、diffusion-projected-ganの中に保存してください
• 以下のコマンドで画像を生成します
  • python gen_images.py --outdir=out --seeds=0-10 --network=./best_model.pkl
  • パッケージがインストールされていないというエラーが出たら適宜インストールしてください
• outフォルダに生成された画像が入っています

商用利用などはご遠慮ください