mayrajeo commited on
Commit
b04b2b6
1 Parent(s): c27637d

Update src/yolov8_onnx.py

Browse files
Files changed (1) hide show
  1. 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
- providers = ['CUDAExecutionProvider', 'CPUExecutionProvider']
35
- else:
36
- providers = ['CPUExecutionProvider']
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,