From 31246d927c02597e4b0bda37ce7f76bd32ccfcc9 Mon Sep 17 00:00:00 2001
From: Elijah Harmon <elijahharmon@gmail.com>
Date: Wed, 29 Mar 2023 02:45:20 -0400
Subject: [PATCH] Update main_onnx_amd.py

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

diff --git a/main_onnx_amd.py b/main_onnx_amd.py
index 5ac6df8..1b76191 100644
--- a/main_onnx_amd.py
+++ b/main_onnx_amd.py
@@ -116,7 +116,7 @@ def main():
         if len(im.shape) == 3:
             im = im[None]
 
-        outputs = ort_sess.run(None, {'images': im})
+        outputs = ort_sess.run(None, {'images': np.array(im)})
 
         im = torch.from_numpy(outputs[0]).to('cpu')