Skip to content

Commit ae885d8

Browse files
committed
hotfix(seflow): typo on dynamic chamfer loss
1 parent 6f20f8f commit ae885d8

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

scripts/network/loss_func.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ def seflowLoss(res_dict, timer=None):
4949
# timer[5][2].start("DynamicChamferDistance")
5050
dynamic_chamfer_dis = torch.tensor(0.0, device=est_flow.device)
5151
if have_dynamic_cluster:
52-
dynamic_chamfer_dis += MyCUDAChamferDis(pc0_dynamic, pc1_dynamic, truncate_dist=TRUNCATED_DIST)
52+
dynamic_chamfer_dis += MyCUDAChamferDis(pseudo_pc1from0[pc0_label>0], pc1_dynamic, truncate_dist=TRUNCATED_DIST)
5353
# timer[5][2].stop()
5454

5555
# third item loss: exclude static points' flow

0 commit comments

Comments
 (0)