面对来自 Pixel Nerf 模型的 AssertionError

Question

I am using this model from github, Form this library we can generate a video from 1,2 or 3 images. By Running this command (!python eval/gen_video.py -n srn_car --split test -P '64 104' -D "/content/drive/MyDrive/pixel-nerf/test/000000.png" -S 1)

link

上给出的实际命令

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.png
Loading SRN dataset /content/drive/MyDrive/pixel-nerf/test/000000.png_test name: 000000.png
Traceback (most recent call last):
File "eval/gen_video.py", line 69, in <module>
args.dataset_format, args.datadir, want_split=args.split, training=False
File "/content/drive/My Drive/pixel-nerf/src/data/__init__.py", line 62, in get_split_dataset
test_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

Answer 1

以防万一您仍然对上述问题感兴趣。 -D 代表数据集目录而不是图像位置。因此，根据 Github link 中给出的示例，文件夹结构如下：

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

面对来自 Pixel Nerf 模型的 AssertionError

Facing an AssertionError in from a Pixel Nerf model

machine-learning

computer-vision

deep-learning

data-science