We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 6f20f8f commit ae885d8Copy full SHA for ae885d8
1 file changed
scripts/network/loss_func.py
@@ -49,7 +49,7 @@ def seflowLoss(res_dict, timer=None):
49
# timer[5][2].start("DynamicChamferDistance")
50
dynamic_chamfer_dis = torch.tensor(0.0, device=est_flow.device)
51
if have_dynamic_cluster:
52
- dynamic_chamfer_dis += MyCUDAChamferDis(pc0_dynamic, pc1_dynamic, truncate_dist=TRUNCATED_DIST)
+ dynamic_chamfer_dis += MyCUDAChamferDis(pseudo_pc1from0[pc0_label>0], pc1_dynamic, truncate_dist=TRUNCATED_DIST)
53
# timer[5][2].stop()
54
55
# third item loss: exclude static points' flow
0 commit comments