我正在使用这个来自GitHub的模型,使用这个库,我们可以从1、2或3张图片生成视频。通过运行这个命令
(!python eval/gen_video.py -n srn_car --split test -P '64 104' -D "/content/drive/MyDrive/pixel-nerf/test/000000.png" -S 1)
链接中提供的实际命令
python eval/gen_video.py -n srn_car --gpu_id=<GPU (s)> --split test -P '64 104' -D <srn data dir>/cars -S
我删除了gpu_id,因为我在Google Colab上运行代码。
这是我放在测试目录中的图片的路径。(/content/drive/MyDrive/pixel-nerf/test/000000.png”)
我遇到的错误
EXPERIMENT NAME: srn_car* Config file: conf/exp/srn.conf* Dataset format: srn* Dataset location: /content/drive/MyDrive/pixel-nerf/test/000000.pngLoading SRN dataset /content/drive/MyDrive/pixel-nerf/test/000000.png_test name: 000000.pngTraceback (most recent call last):File "eval/gen_video.py", line 69, in <module>args.dataset_format, args.datadir, want_split=args.split, training=FalseFile "/content/drive/My Drive/pixel-nerf/src/data/__init__.py", line 62, in get_split_datasettest_set = dset_class(datadir, stage="test", **flags, **kwargs)File "/content/drive/My Drive/pixel-nerf/src/data/SRNDataset.py", line 29, in __init__assert os.path.exists(self.base_path)AssertionError
回答:
如果你仍然对上面的问题感兴趣,以防万一。-D表示数据集目录,而不是图片位置。因此,根据GitHub链接中给出的示例,文件夹结构如下:
srn_dataset/chairs_train
srn_dataset/chairs_eval
srn_dataset/chairs_test
因此,评估的语法将是:python eval/gen_video.py -n srn_chair –split test -P ’64 104′ -D srn_dataset/chairs -S 1