Questions about OrientedRepPointsHead.loss()

[crisz94]

Hi @LiWentomng , thanks for your execellent work!

In orientedreppoints_head.py, line 371 to line 381 seems to compute proposals to get targets of refine stage. But the computation of proposals are different from the funtion offset_to_pts(). More specifically, offset_to_pts() use pts_shift as yx_pts_shift, but line 371 to line 381 in orientedreppoints_head.py uses points_shift as xy_pts_shift.

Consequently, pts_preds_init is used differently in forward() and loss() and this may lead to some unpredicted results. Did I misunderstand your code? It's really confused, hope to get your respond soon!