Monocular, One-stage, Regression of Multiple 3D People and their 3D positions & trajectories in camera & global coordinates. ROMP[ICCV21], BEV[CVPR22], TRACE[CVPR2023]
INFO:root:Training all layers.
Initialization finished!
INFO:root:gathering single image datasets.
INFO:root:CMU Panoptic dataset total 9600 samples, loading test split
Traceback (most recent call last):
File "/xxx/miniconda3/envs/torch10/lib/python3.9/runpy.py", line 197, in _run_module_as_main
return _run_code(code, main_globals, None,
File "/xxx/miniconda3/envs/torch10/lib/python3.9/runpy.py", line 87, in _run_code
exec(code, run_globals)
File "/xxx/ROMP/romp/test.py", line 204, in
main()
File "/xxx/ROMP/romp/test.py", line 195, in main
demo.test_cmu_panoptic()
File "/xxx/ROMP/romp/test.py", line 53, in test_cmu_panoptic
for test_iter,meta_data in enumerate(data_loader):
File "/xxx/miniconda3/envs/torch10/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 521, in next
data = self._next_data()
File "/xxx/miniconda3/envs/torch10/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 561, in _next_data
data = self._dataset_fetcher.fetch(index) # may raise StopIteration
File "/xxx/miniconda3/envs/torch10/lib/python3.9/site-packages/torch/utils/data/_utils/fetch.py", line 49, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/xxx/miniconda3/envs/torch10/lib/python3.9/site-packages/torch/utils/data/_utils/fetch.py", line 49, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/xxx/ROMP/romp/lib/models/../dataset/mixed_dataset.py", line 101, in getitem
return self.dataset[index]
File "/xxx/ROMP/romp/lib/models/../dataset/image_base.py", line 487, in getitem
return self.get_item_single_frame(index)
File "/xxx/ROMP/romp/lib/models/../dataset/image_base.py", line 117, in get_item_single_frame
kp3d, valid_masks[:,1] = self.process_kp3ds(info['kp3ds'], used_person_inds, \
File "/xxx/ROMP/romp/lib/models/../dataset/image_base.py", line 359, in process_kp3ds
kp3d_processed[inds] = kp3d
ValueError: could not broadcast input array from shape (17,3) into shape (54,3)
INFO:root:Training all layers. Initialization finished! INFO:root:gathering single image datasets. INFO:root:CMU Panoptic dataset total 9600 samples, loading test split Traceback (most recent call last): File "/xxx/miniconda3/envs/torch10/lib/python3.9/runpy.py", line 197, in _run_module_as_main return _run_code(code, main_globals, None, File "/xxx/miniconda3/envs/torch10/lib/python3.9/runpy.py", line 87, in _run_code exec(code, run_globals) File "/xxx/ROMP/romp/test.py", line 204, in
main()
File "/xxx/ROMP/romp/test.py", line 195, in main
demo.test_cmu_panoptic()
File "/xxx/ROMP/romp/test.py", line 53, in test_cmu_panoptic
for test_iter,meta_data in enumerate(data_loader):
File "/xxx/miniconda3/envs/torch10/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 521, in next
data = self._next_data()
File "/xxx/miniconda3/envs/torch10/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 561, in _next_data
data = self._dataset_fetcher.fetch(index) # may raise StopIteration
File "/xxx/miniconda3/envs/torch10/lib/python3.9/site-packages/torch/utils/data/_utils/fetch.py", line 49, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/xxx/miniconda3/envs/torch10/lib/python3.9/site-packages/torch/utils/data/_utils/fetch.py", line 49, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/xxx/ROMP/romp/lib/models/../dataset/mixed_dataset.py", line 101, in getitem
return self.dataset[index]
File "/xxx/ROMP/romp/lib/models/../dataset/image_base.py", line 487, in getitem
return self.get_item_single_frame(index)
File "/xxx/ROMP/romp/lib/models/../dataset/image_base.py", line 117, in get_item_single_frame
kp3d, valid_masks[:,1] = self.process_kp3ds(info['kp3ds'], used_person_inds, \
File "/xxx/ROMP/romp/lib/models/../dataset/image_base.py", line 359, in process_kp3ds
kp3d_processed[inds] = kp3d
ValueError: could not broadcast input array from shape (17,3) into shape (54,3)
hwo to fix it?