StyleGan2 ADA PyTorch 输出图像分辨率断言错误

Question

我正在尝试使用 StyleGan2 生成图像。

正在执行此单元格：

!python train.py --gpus=1 --cfg=$config --metrics=None --outdir=./results --data=$dataset_path --snap=$snapshot_count --resume=$resume_from --augpipe=$augs --initstrength=$aug_strength --gamma=$gamma_value --mirror=$mirror_x --mirrory=False --nkimg=$train_count

输出如下：

Output directory:   ./results/00001-datasets-mirror-11gb-gpu-gamma50-bg-resumecustom
Training data:      /content/datasets.zip
Training duration:  25000 kimg
Number of GPUs:     1
Number of images:   608
Image resolution:   600
Conditional model:  False
Dataset x-flips:    True

Creating output directory...
Launching processes...
Loading training set...

Num images:  1216
Image shape: [3, 600, 600]
Label shape: [0]

然后给我下面的断言错误：

assert img_resolution >= 4 and img_resolution & (img_resolution - 1) == 0
Assertion Error:

我检查了错误的source。但说句公道话，我没看懂。

Answer 1

我发现：问题是图像的输入尺寸必须至少 [3, 1024, 1024] 我的是 [3, 600, 600]

StyleGan2 ADA PyTorch 输出图像分辨率断言错误

StyleGan2 ADA PyTorch Output Image Resolution Assertion Error

python

pytorch

stylegan