wrong index in joint_loss?

[Droliven]

In your line 20 of the train.py joint_loss, it writes:

parts_idx = [(np.array(p) * 3).tolist() + (np.array(p) * 3 + 1).tolist() + (np.array(p) * 3 + 2).tolist() for p in parts]

I think the 48 joints are in order of 16 x 3, instead of 3 x 16, so parts_idx should be:

parts_idx = [np.asarray([[i*3, i*3+1, i*3+2] for i in p]).flatten().tolist() for p in parts]

[XiaoyuBIE1994]

I think that won't change the results when computing the loss