Giters

apple / ml-neuman

Official repository of NeuMan: Neural Human Radiance Field from a Single Video (ECCV 2022)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Error with visualize_scene.py -- cannot advance beyond assert check

domattioli opened this issue · comments

commented

Preprocess script for a custom video passed all ten steps of the bash script. There are reasonable smpl overlays in the overlays/optimized and overlays/romp folders. Running

python visualize_scene.py --scene_dir /home/me/src/ml-neuman/data/example-single_person-moving-football_receiver_wide_open/my_video/output

produces the error traceback:

entered main
reading cameras: 100%|████████████████████████████████████████████████████████████████████████████████████████████| 1/1 [00:00<00:00, 29959.31it/s]
reading images meta: 100%|████████████████████████████████████████████████████████████████████████████████████████| 9/9 [00:00<00:00, 12002.78it/s]
reading point cloud: 100%|███████████████████████████████████████████████████████████████████████████████████| 359/359 [00:00<00:00, 423440.70it/s]
Computing near/far for ['bkg']: 100%|██████████████████████████████████████████████████████████████████████████████| 9/9 [00:00<00:00, 5588.27it/s]

...creating 'smpl_path' var...
assert command will check this path: /home/me/src/ml-neuman/data/smplx/smpl/SMPL_NEUTRAL.pkl
....advancing to assert check...

using optimized smpl
Traceback (most recent call last):
  File "visualize_scene.py", line 64, in <module>
    main(opt)
  File "visualize_scene.py", line 28, in main
    smpl_type='optimized'
  File "/home/me/src/ml-neuman/data_io/neuman_helper.py", line 245, in read_scene
    smpls, world_verts, static_verts, Ts = cls.read_smpls(scene_dir, scene.captures, scale=scale, smpl_type=smpl_type)
  File "/home/me/src/ml-neuman/data_io/neuman_helper.py", line 281, in read_smpls
    raw_smpl = joblib.load(smpl_path)
  File "/home/me/venv/neuman_venv/lib/python3.7/site-packages/joblib/numpy_pickle.py", line 587, in load
    obj = _unpickle(fobj, filename, mmap_mode)
  File "/home/me/venv/neuman_venv/lib/python3.7/site-packages/joblib/numpy_pickle.py", line 506, in _unpickle
    obj = unpickler.load()
  File "/home/me/venv/neuman_venv/lib/python3.7/pickle.py", line 1088, in load
    dispatch[key[0]](self)
KeyError: 0

The file extension is .pkl, not .pk1, but to me, the KeyError: 0 suggests that this is an issue with the python 3.7 pickle.py function. Here is a screenshot of my output folder:
image