guided-diffusion を Google Colaboratory で実装したい

前提・実現したいこと

下記の Diffusion model を Google Colaboratory で試しています。(初学者です)
https://github.com/openai/guided-diffusion

README に従い models フォルダの作成、そこに学習済みモデルのチェックポイントのダウンロードまで終了しています。

発生している問題・エラーメッセージ

学習済みモデルからのサンプリングを行いたいのですが、コードを実行すると下記のエラーメッセージが出ます。

python
1from guided_diffusion import dist_util, logger
2ModuleNotFoundError: No module named 'guided_diffusion'

該当のソースコード

python
1!git clone https://github.com/openai/guided-diffusion.git
2
3%cd /content/guided-diffusion
4!mkdir /content/guided-diffusion/models
5
6# 学習済みモデルのダウンロード
7%cd /content/guided-diffusion/models
8
9!wget https://openaipublic.blob.core.windows.net/diffusion/jul-2021/256x256_diffusion.pt
10
11!wget https://openaipublic.blob.core.windows.net/diffusion/jul-2021/256x256_diffusion_uncond.pt
12
13!wget https://openaipublic.blob.core.windows.net/diffusion/jul-2021/256x256_classifier.pt
14
15%cd ..
16
17!pip install .
18
19%cd /content/guided-diffusion/scripts
20
21# 学習済みモデルからのサンプリング
22!python classifier_sample.py
23
24!python image_sample.py
25
26!python super_res_sample.py
27
28SAMPLE_FLAGS="--batch_size 4 --num_samples 100 --timestep_respacing 250"

補足情報（FW/ツールのバージョンなど）

jbpb0

2022/11/01 07:47

質問に記載のエラーの箇所に到達する前に、別のエラーが発生します > %cd /content/guided-diffusion/models を実行すると、「[Errno 2] No such file or directory: '/content/guided-diffusion/models'」というエラーが発生します https://github.com/openai/guided-diffusion には「models」というディレクトリが無いので、 > !git clone https://github.com/openai/guided-diffusion.git を実行した後、「/content/guided-diffusion」に「models」というディレクトリが無いためですそのエラーを無視して先に進むと、 > %cd .. でカレントディレクトリが「/」になり、そこで > !pip install . を実行すると、「ERROR: Directory '.' is not installable. Neither 'setup.py' nor 'pyproject.toml' found.」というエラーが発生します質問者さんが実行したらそうならないのなら、質問に記載のコードと実際に実行してるコードが異なってるのだと思いますできるだけ、実際に実行してるコードを質問に記載してください全く同じではなくてもいいですが、質問に記載のエラーの箇所に到達する前に別のエラーが発生するようなのは、やめてください質問のエラーが再現できないため、質問内容を検討することができません質問者さんと同じ状態で、質問のエラーを他人が再現できるように、質問は書いてください

riku_university

2022/11/01 08:12

前提・実現したいことのほうに「README に従い models フォルダの作成、そこに学習済みモデルのチェックポイントのダウンロードまで終了しています。」と記載しましたが該当のソースコードのほうに記載するべきでした。 models ディレクトリは git clone を行ったあとに手動で作成したので該当のソースコード内に記載していませんでした。コードを追記しておきます。

jbpb0

2022/11/03 01:49

私の回答のように実行したら、この質問のエラー > ModuleNotFoundError: No module named 'guided_diffusion' は出なくなるはずですが、いかがでしょうか？

行動規範の内容に同意します

回答1件

ベストアンサー

質問のエラー

ModuleNotFoundError: No module named 'guided_diffusion'

は、

python
1!python classifier_sample.py
2!python image_sample.py
3!python super_res_sample.py

↓ 変更

python
1!PYTHONPATH=/content/guided-diffusion:$PYTHONPATH python classifier_sample.py
2!PYTHONPATH=/content/guided-diffusion:$PYTHONPATH python image_sample.py
3!PYTHONPATH=/content/guided-diffusion:$PYTHONPATH python super_res_sample.py