Thanks for your greate work. And when I study the code, I find a re-weight, i.e. gt_pts_weights, is used for loss_pts between gt_pts and gt_paired_pts. But there is no re-weight for loss_pts between pred_pts and pred_paired_pts. Does it affect the performance?
# calculate loss pts
loss_pts = pred_pts.new_tensor(0)
loss_pts += self.loss_pts(gt_pts,
gt_paired_pts,
weight=gt_pts_weights[..., None],
avg_factor=gt_pts.shape[0])
loss_pts += self.loss_pts(pred_pts,
pred_paired_pts,
avg_factor=pred_pts.shape[0])
Thanks for your greate work. And when I study the code, I find a re-weight, i.e. gt_pts_weights, is used for loss_pts between gt_pts and gt_paired_pts. But there is no re-weight for loss_pts between pred_pts and pred_paired_pts. Does it affect the performance?