uncommenting lines that are actually important ;P

faustomilletari · Jun 13, 2016 · fff97b9 · fff97b9
1 parent 9b3a020
commit fff97b9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/VNet.py b/VNet.py
@@ -165,7 +165,7 @@ def train(self):
 
     def test(self):
         self.dataManagerTest = DM.DataManager(self.params['ModelParams']['dirTest'], self.params['ModelParams']['dirResult'], self.params['DataManagerParams'])
-        #self.dataManagerTest.loadTestData()
+        self.dataManagerTest.loadTestData()
 
         net = caffe.Net(self.params['ModelParams']['prototxtTest'],
                         os.path.join(self.params['ModelParams']['dirSnapshots'],"_iter_" + str(self.params['ModelParams']['snapshot']) + ".caffemodel"),
@@ -194,5 +194,5 @@ def test(self):
 
             results[key] = np.squeeze(labelmap)
 
-            #self.dataManagerTest.writeResultsFromNumpyLabel(np.squeeze(labelmap),key)
+            self.dataManagerTest.writeResultsFromNumpyLabel(np.squeeze(labelmap),key)