ValueError: could not broadcast input array from shape (2,15) into shape (2,16)

Question

ValueError: could not broadcast input array from shape (2,15) into shape (2,16)

liusui652 opened this issue 3 years ago · 3 comments

When I ran train.py, this bug appeared. Later, I adjusted the torch version and no longer reported the error, but it appeared again when I ran evaluate_model.py. I don't know what caused it?

Answer 1 · 2022-05-12T11:42:45.000Z

@liusui652 I'm having same problem when I run train.py. What torch version did you adjusted?

Answer 2 · 2022-12-25T15:49:07.000Z

Hi @liusui652 ! Did you yolved the problem? I'm currently facing the same issue

Answer 3 · 2023-04-20T17:06:30.000Z

It is because some person is not in all 16 positions (8 observed, 8 predicted) - in your case, "(2,15)" means there are 15 positions of some person in dataset instead of 16. I had the same issue so I added these lines in trajectory.py under line curr_ped_seq = np.transpose(curr_ped_seq[:, 2:]):

if curr_ped_seq.shape != (2,16): continue

(in case you are predicting 12 positions, use (2,20) )