modify

LouisFinner · Mar 21, 2018 · 94c3a0c · 94c3a0c
1 parent f550b37
commit 94c3a0c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/api.py b/api.py
@@ -110,9 +110,9 @@ def process(self, input, image_info = None):
         cropped_image = warp(image, tform.inverse, output_shape=(self.resolution_inp, self.resolution_inp))
 
         # run our net
-        st = time()
+        #st = time()
         cropped_pos = self.net_forward(cropped_image)
-        print 'net time:', time() - st
+        #print 'net time:', time() - st
 
         # restore 
         cropped_pos = np.squeeze(cropped_pos)

diff --git a/run_basics.py b/run_basics.py
@@ -9,7 +9,7 @@
 from utils.write import write_obj
 
 # ---- init PRN
-os.environ['CUDA_VISIBLE_DEVICES'] = '-1' # GPU number
+os.environ['CUDA_VISIBLE_DEVICES'] = '0' # GPU number, -1 for CPU
 prn = PRN(is_dlib = False, is_opencv = False)