hb-setosys commited on
Commit
1cc5861
·
verified ·
1 Parent(s): 99a40c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -7,8 +7,12 @@ import os
7
  net = cv2.dnn.readNet('yolov3.weights', 'yolov3.cfg')
8
 
9
  # Enable GPU (if available)
10
- net.setPreferableBackend(cv2.dnn.DNN_BACKEND_CUDA)
11
- net.setPreferableTarget(cv2.dnn.DNN_TARGET_CUDA)
 
 
 
 
12
 
13
  # Load class names
14
  with open('coco.names', 'r') as f:
 
7
  net = cv2.dnn.readNet('yolov3.weights', 'yolov3.cfg')
8
 
9
  # Enable GPU (if available)
10
+ #net.setPreferableBackend(cv2.dnn.DNN_BACKEND_CUDA)
11
+ #net.setPreferableTarget(cv2.dnn.DNN_TARGET_CUDA)
12
+
13
+ # Set backend to CPU
14
+ net.setPreferableBackend(cv2.dnn.DNN_BACKEND_OPENCV)
15
+ net.setPreferableTarget(cv2.dnn.DNN_TARGET_CPU)
16
 
17
  # Load class names
18
  with open('coco.names', 'r') as f: