diff --git a/main_onnx_cpu.py b/main_onnx_cpu.py
index 20b5259..8fa1407 100644
--- a/main_onnx_cpu.py
+++ b/main_onnx_cpu.py
@@ -117,7 +117,7 @@ def main():
         if len(im.shape) == 3:
             im = im[None]
 
-        outputs = ort_sess.run(None, {'images': npImg})
+        outputs = ort_sess.run(None, {'images': np.array(im)})
 
         im = torch.from_numpy(outputs[0]).to('cpu')