the shape of groud thruth is different the shape of out

leeyegy / SimCC

[ECCV'2022 Oral] PyTorch implementation for: SimCC: a Simple Coordinate Classification Perspective for Human Pose Estimation (http://arxiv.org/abs/2107.03332). Old name: SimDR

317 stars 29 forks source link

the shape of groud thruth is different the shape of out #4

Open zouxuelian opened 2 years ago

zouxuelian commented 2 years ago

loss += self.criterion(coord_x_pred,coord_gt[:,0]).mul(weight).mean()

IndexError: too many indices for tensor of dimension 1