From f3bd0cea5ab2783eab5f407765316b17e195b854 Mon Sep 17 00:00:00 2001 From: Elijah Harmon Date: Tue, 28 Mar 2023 23:42:32 -0700 Subject: [PATCH] Update main_onnx_amd.py (#74) --- 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 aa6d9fb..5ac6df8 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': npImg}) + outputs = ort_sess.run(None, {'images': im}) im = torch.from_numpy(outputs[0]).to('cpu')