Update app.py
Browse files
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:
|