Update src/yolov8_onnx.py
Browse files- src/yolov8_onnx.py +4 -4
src/yolov8_onnx.py
CHANGED
@@ -30,10 +30,10 @@ class Yolov8onnx:
|
|
30 |
self.iou_thres = iou_thres
|
31 |
self.input_width = input_width
|
32 |
self.input_height = input_height
|
33 |
-
if device == 'cpu':
|
34 |
-
|
35 |
-
else:
|
36 |
-
|
37 |
|
38 |
self.onnx_session = ort.InferenceSession(
|
39 |
onnx_model,
|
|
|
30 |
self.iou_thres = iou_thres
|
31 |
self.input_width = input_width
|
32 |
self.input_height = input_height
|
33 |
+
#if device == 'cpu':
|
34 |
+
# providers = ['CUDAExecutionProvider', 'CPUExecutionProvider']
|
35 |
+
#else:
|
36 |
+
providers = ['CPUExecutionProvider']
|
37 |
|
38 |
self.onnx_session = ort.InferenceSession(
|
39 |
onnx_model,
|