From 55955f77c363312dcaee8a3ef7b501512c63f822 Mon Sep 17 00:00:00 2001
From: "Dr.Lt.Data" <dr.lt.data@gmail.com>
Date: Wed, 24 May 2023 09:11:23 +0900
Subject: [PATCH] onnx error fix

---
 onnx.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/onnx.py b/onnx.py
index d687a22..ddeedff 100644
--- a/onnx.py
+++ b/onnx.py
@@ -15,7 +15,7 @@ try:
         image -= [103.939, 116.779, 123.68]  # 'caffe' mode image preprocessing
 
         # do detection
-        onnx_model = onnxruntime.InferenceSession(onnx_model)
+        onnx_model = onnxruntime.InferenceSession(onnx_model, providers=["CPUExecutionProvider"])
         outputs = onnx_model.run(
             [s_i.name for s_i in onnx_model.get_outputs()],
             {onnx_model.get_inputs()[0].name: np.expand_dims(image, axis=0)},