Skip to content

Commit 072486b

Browse files
authored
fix: remove func didn't use in deflow.
1 parent 6723b91 commit 072486b

1 file changed

Lines changed: 1 addition & 3 deletions

File tree

scripts/network/loss_func.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,6 @@
88
99
"""
1010
import torch
11-
from assets.cuda.chamfer_cuda import nnChamferDis
12-
MyCUDAChamferDistance = nnChamferDis()
1311

1412
def deflowLoss(res_dict):
1513
pred = res_dict['est_flow']
@@ -132,4 +130,4 @@ def evaluate_leaderboard(est_flow, rigid_flow, pc0, gt_flow, is_valid, pts_ids):
132130
if np.isnan(EPE_Dynamic) or np.isnan(Dynamic_IoU) or np.isnan(EPE_Background_Static) or np.isnan(EPE_Foreground_Static):
133131
print(res_dict)
134132

135-
return EPE_Background_Static, EPE_Dynamic, EPE_Foreground_Static, Dynamic_IoU
133+
return EPE_Background_Static, EPE_Dynamic, EPE_Foreground_Static, Dynamic_IoU

0 commit comments

Comments
 (0)