Skip to content

confusion about the reweight for CD loss #2

@drilistbox

Description

@drilistbox

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])

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions