From 9015b9161db594d4650bfacb8aea49c36b7befc9 Mon Sep 17 00:00:00 2001
From: Haozhe Qi <32994368+HaozheQi@users.noreply.github.com>
Date: Mon, 9 Mar 2020 14:53:36 +0800
Subject: [PATCH] Update test_tracking.py

---
 test_tracking.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/test_tracking.py b/test_tracking.py
index 8a8e4c6..3a1e045 100644
--- a/test_tracking.py
+++ b/test_tracking.py
@@ -80,7 +80,7 @@ def test(loader,model,epoch=-1,shape_aggregation="",reference_BB="",model_fusion
                                         offset=dataset.offset_BB,
                                         scale=dataset.scale_BB)
                         
-                        candidate_PCs,candidate_labels,candidate_reg = utils.regularizePCwithlabel(candidate_PC, candidate_label,candidate_reg,dataset.input_size)
+                        candidate_PCs,candidate_labels,candidate_reg = utils.regularizePCwithlabel(candidate_PC, candidate_label,candidate_reg,dataset.input_size,istrain=False)
                         
                         candidate_PCs_torch = candidate_PCs.unsqueeze(0).cuda()
 
@@ -96,7 +96,7 @@ def test(loader,model,epoch=-1,shape_aggregation="",reference_BB="",model_fusion
                         else:
                             model_PC = utils.getModel(PCs[:i],results_BBs,offset=dataset.offset_BB,scale=dataset.scale_BB)
 
-                        model_PC_torch = utils.regularizePC(model_PC, dataset.input_size).unsqueeze(0)
+                        model_PC_torch = utils.regularizePC(model_PC, dataset.input_size,istrain=False).unsqueeze(0)
                         model_PC_torch = Variable(model_PC_torch, requires_grad=False).cuda()
                         candidate_PCs_torch = Variable(candidate_PCs_torch, requires_grad=False).cuda()