使用generate.py StyleGAN 2 ADA时无法识别的参数

Question

我已经用Google Colab中的自定义乳房X光照片数据集训练了StyleGAN 2ADA (link to their repo)的ffqh1024模型。我训练过的模型.pkl文件已准备好放在驱动器文件夹中，我希望使用该.pkl文件生成图像。我试过了：

!python generate.py --outdir='/content/drive/MyDrive/TFM/Generated' --trunc=1 --seeds=85,265,297,849 \ --network='/content/drive/MyDrive/TFM/colab-sg2-ada/stylegan2-ada/results/00025-ddsm-auto1-bg-resumecustom/network-snapshot-000096.pkl'

就像在GitHub上推荐的一样，但我得到了这个错误：

usage: generate.py [-h] {generate-images,truncation-traversal,generate-latent-walk,generate-neighbors,lerp-video} ... generate.py: error: unrecognized arguments: --outdir='/content/drive/MyDrive/TFM/Generated' --trunc=1 --seeds=85,265,297,849 --network='/content/drive/MyDrive/TFM/colab-sg2-ada/stylegan2-ada/results/00025-ddsm-auto1-bg-resumecustom/network-snapshot-000096.pkl'

我真的不知道为什么generate.py不能识别参数...我不得不!pip安装opensimplex才能让generate.py运行，不知道是不是跟这个问题有关…

在StyleGAN 2ADA存储库中，有一些使用经过训练的模型生成图像的示例：

# Generate curated MetFaces images without truncation (Fig.10 left)
python generate.py --outdir=out --trunc=1 --seeds=85,265,297,849 \
    --network=https://nvlabs-fi-cdn.nvidia.com/stylegan2-ada/pretrained/metfaces.pkl

这是执行!python generate.py -h的结果：

usage: generate.py [-h]
                   {generate-images,truncation-traversal,generate-latent-walk,generate-neighbors,lerp-video}
                   ...

Generate images using pretrained network pickle.

positional arguments:
  {generate-images,truncation-traversal,generate-latent-walk,generate-neighbors,lerp-video}
                        Sub-commands
    generate-images     Generate images
    truncation-traversal
                        Generate truncation walk
    generate-latent-walk
                        Generate latent walk
    generate-neighbors  Generate random neighbors of a seed
    lerp-video          Generate interpolation video (lerp) between random
                        vectors

optional arguments:
  -h, --help            show this help message and exit

examples:

  # Generate curated MetFaces images without truncation (Fig.10 left)
  python generate.py --outdir=out --trunc=1 --seeds=85,265,297,849 \
      --network=https://nvlabs-fi-cdn.nvidia.com/stylegan2-ada/pretrained/metfaces.pkl

  # Generate uncurated MetFaces images with truncation (Fig.12 upper left)
  python generate.py --outdir=out --trunc=0.7 --seeds=600-605 \
      --network=https://nvlabs-fi-cdn.nvidia.com/stylegan2-ada/pretrained/metfaces.pkl

  # Generate class conditional CIFAR-10 images (Fig.17 left, Car)
  python generate.py --outdir=out --trunc=1 --seeds=0-35 --class=1 \
      --network=https://nvlabs-fi-cdn.nvidia.com/stylegan2-ada/pretrained/cifar10.pkl

  # Render image from projected latent vector
  python generate.py --outdir=out --dlatents=out/dlatents.npz \
      --network=https://nvlabs-fi-cdn.nvidia.com/stylegan2-ada/pretrained/ffhq.pkl

Answer 1

尝试从命令行中删除\。我不确定它会不会起作用