Spaces:
Build error
Build error
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:
|