Nvidiaのfew-shot-vid2vidを使うとimportエラー

Question

NvidiaのDeeplearningソフトウェアのfew-shot-vid2vidは、少ない画像で他の画像の真似をさせることができます。例えば、1枚のモナリザの画像に、自分の顔と同じ変形をさせるといったことができます。
昨今、フェイク画像の問題が話題になっており、私も興味津々なので、一度使ってみようと思ったところです。

そのリポジトリは次のURLです。
https://github.com/NVlabs/few-shot-vid2vid

そこで、リポジトリをクローンして、必要なライブラリをインストールして、さてtrainを実行しようとしたら、次のエラーを生じました。

~/few-shot-vid2vid$ python3 train.py --name pose --dataset_mode fewshot_pose --adaptive_spade --warp_ref --spade_combine --remove_face_labels --add_face_D --niter_single 100 --niter 200 --batchSize 2
Traceback (most recent call last):
  File "train.py", line 13, in <module>
    from models.models import create_model
  File "/home/****/few-shot-vid2vid/models/models.py", line 11, in <module>
    from models.networks.sync_batchnorm import DataParallelWithCallback
  File "/home/****/few-shot-vid2vid/models/networks/__init__.py", line 13, in <module>
    from models.networks.discriminator import *
  File "/home/****/few-shot-vid2vid/models/networks/discriminator.py", line 14, in <module>
    from models.networks.architecture import actvn as actvn
  File "/home/****/few-shot-vid2vid/models/networks/architecture.py", line 13, in <module>
    from models.networks.normalization import SPADE, SynchronizedBatchNorm2d
  File "/home/****/few-shot-vid2vid/models/networks/normalization.py", line 15, in <module>
    from apex.parallel import SyncBatchNorm as SynchronizedBatchNorm2d
  File "/home/****/.local/lib/python3.8/site-packages/apex/__init__.py", line 13, in <module>
    from pyramid.session import UnencryptedCookieSessionFactoryConfig
ImportError: cannot import name 'UnencryptedCookieSessionFactoryConfig' from 'pyramid.session' (unknown location)

次に、testを実行しようとしても、同じエラーがでました。

~/few-shot-vid2vid$ python3 test.py --name face --dataset_mode fewshot_face --adaptive_spade --warp_ref --spade_combine --seq_path ./datasets/face/test_images/0001 --ref_img_path ./datasets/face/test_images/0002
Traceback (most recent call last):
  File "test.py", line 15, in <module>
    from models.models import create_model
  File "/home/****/few-shot-vid2vid/models/models.py", line 11, in <module>
    from models.networks.sync_batchnorm import DataParallelWithCallback
  File "/home/****/few-shot-vid2vid/models/networks/__init__.py", line 13, in <module>
    from models.networks.discriminator import *
  File "/home/****/few-shot-vid2vid/models/networks/discriminator.py", line 14, in <module>
    from models.networks.architecture import actvn as actvn
  File "/home/****/few-shot-vid2vid/models/networks/architecture.py", line 13, in <module>
    from models.networks.normalization import SPADE, SynchronizedBatchNorm2d
  File "/home/****/few-shot-vid2vid/models/networks/normalization.py", line 15, in <module>
    from apex.parallel import SyncBatchNorm as SynchronizedBatchNorm2d
  File "/home/****/.local/lib/python3.8/site-packages/apex/__init__.py", line 13, in <module>
    from pyramid.session import UnencryptedCookieSessionFactoryConfig
ImportError: cannot import name 'UnencryptedCookieSessionFactoryConfig' from 'pyramid.session' (unknown location)

このエラー、いくら検索しても解決策がわかりません。このモデルはこれまでかなり利用されてきたものなので、絶対に解決策があるはずです。
どなたかわかる方いらっしゃいましたら、教えてください。
よろしくお願いします。

環境は次のとおりです。

utuntu18.04
Python3.8
pyramid2.0

Answer 1

今遭遇されている問題ですが、エラーを吐いているライブラリであるapexのissueでも言及されている通り、おそらく同名の別ライブラリをインストールされてしまっているために発生しているものと思われます。
https://github.com/NVIDIA/apex/issues/170#issuecomment-495256877

ライブラリインストール時に、

pip install apex

などと実行されている場合、該当します。

対処は https://github.com/NVIDIA/apex#linux に書かれている通り、GitHubのリポジトリから直接インストールする、というものになります。
(現時点では、PyTorch extensionであるapexのPyPIパッケージは配布されていないようです)
例えば上記のページでは以下のように記載されています。

git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

Comments

[good_morning]

ご回答ありがとうございました。
上記のURLのRequirementsには「CUDA 9 or newer」とありますが、小職のマシンはその条件を満たしていませんので、また改めてチャレンジしてみたいと思います。
それでは、よろしくお願いします。