Error with visualize.py #48
Description
Hey!
I had a problem running the file visualize.py. I could succesfully run, though had to do some modifications to avoid warnings, the codes "train.py" and "test.py" as I am using the Social LSTM model.
However, when I had to actually execute 'visualize.py', I got this error:
Now processing: nexus_0_3.pkl
Folder succesfully removed: ./plot/SOCIALLSTM/LSTM/videos/nexus_0_3
Folder succesfully removed: ./plot/SOCIALLSTM/LSTM/plots/nexus_0_3
ERROR: setting an array element with a sequence. The requested array has an inhomogeneous shape after 2 dimensions. The detected shape was (131, 6) + inhomogeneous part.
I don't know why this line is giving problems: result_arr = np.array(results)
Do you know something that could help me solve this issue?
Thank you in advance!