lixiny / poem Goto Github PK

Hi Lixiny，

Thanks for your great work.
I have tried to reproduce the valuate result as your guided.
$ python scripts/eval.py --cfg config/release/${MODEL}_${DATASET}.yaml -g 0 -b 8 --reload ${PATH_TO_CKPT}

My valuated cmd based on DexYCBMV dataset:
python scripts/eval.py --cfg config\release\POEM_DexYCBMV.yaml -g 0 -b 8 --reload checkpoints\POEM_DexYCBMV-20230912T110921Z-001\POEM_DexYCBMV\checkpoint\PtEmbedMultiviewStereo.pth.tar --eval_draw draw

The error is listed below, could you help to point out the root cause?

Traceback (most recent call last):
File "scripts/eval.py", line 96, in
main_worker(cfg, arg, exp_time)
File "scripts/eval.py", line 73, in main_worker
for bidx, batch in enumerate(testbar):
File "C:\Users\asus.conda\envs\POEM2\lib\site-packages\tqdm\std.py", line 1182, in iter
for obj in iterable:
File "C:\Users\asus.conda\envs\POEM2\lib\site-packages\torch\utils\data\dataloader.py", line 530, in next
data = self._next_data()
File "C:\Users\asus.conda\envs\POEM2\lib\site-packages\torch\utils\data\dataloader.py", line 1224, in _next_data
return self._process_data(data)
File "C:\Users\asus.conda\envs\POEM2\lib\site-packages\torch\utils\data\dataloader.py", line 1250, in _process_data
data.reraise()
File "C:\Users\asus.conda\envs\POEM2\lib\site-packages\torch_utils.py", line 457, in reraise
raise exception
AssertionError: Caught AssertionError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "C:\Users\asus.conda\envs\POEM2\lib\site-packages\torch\utils\data_utils\worker.py", line 287, in _worker_loop
data = fetcher.fetch(index)
File "C:\Users\asus.conda\envs\POEM2\lib\site-packages\torch\utils\data_utils\fetch.py", line 49, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "C:\Users\asus.conda\envs\POEM2\lib\site-packages\torch\utils\data_utils\fetch.py", line 49, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "d:\zhanxiaopan\handtracking\poem\poem\lib\datasets\dexycb.py", line 398, in getitem
assert const_v_id != -101, f"Cannot find the constant camera serial {self.CONST_CAM_SERIAL} in the dataset"
AssertionError: Cannot find the constant camera serial 840412060917 in the dataset

To test on custom data, what are the requirements for calibration added to dataset ?