search

iSEE-Laboratory / PPT

(ECCV 2024) Official PyTorch implementation of paper "Progressive Pretext Task Learning for Human Trajectory Prediction"
MIT License
24 stars 4 forks source link

train_PPT problem #3

Open subetter opened 3 weeks ago

subetter commented 3 weeks ago

Hello, Many thanks for your excellent work! I am currently attempting to reproduce your results.

When I execute run.sh, I encounter the following error: image After debugging, it seems that the sublists within traj_new have inconsistent lengths, which prevents it from being converted into a NumPy array. image

Could you please provide guidance on how to resolve this issue? Thank you very much!

Xiaotong-Lin commented 1 week ago

Hi @subetter , Thanks for your interest! You can try adding dtype=object to the np.array() function on lines 82, 83, and 87 in dataset_loader.py. For example, change np.array(traj_new) to np.array(traj_new, dtype=object). This should resolve the issue.