search

princeton-vl / pose-ae-train

Training code for "Associative Embedding: End-to-End Learning for Joint Detection and Grouping"
BSD 3-Clause "New" or "Revised" License
373 stars 76 forks source link

Some questions about the x and y coordinates! #44

Open FightStone opened 4 years ago

FightStone commented 4 years ago
  1. In calc() function which in group.py file, x and y define for this: x = ind % w y = (ind / w).long() ind_k = torch.stack((x, y), dim=3)
  2. But in adjust() function which in group.py file, the order of x and y is: y, x = joint[0:2]
  3. And in refine() function which in test.py file, the order of x and y is: y, x = keypoints[i][:2].astype(np.int32)
  4. Bue in genDtByPred() function which in test.py file, order of x and y is: tmp["keypoints"] += [float(j[0]), float(j[1]), 1]
FightStone commented 4 years ago

Who can help me answer the question? and Is the code correct?？ Thanks a lot !