From b59317a3ffcd2512e1fc1413fc4ad64ef9f6b0af Mon Sep 17 00:00:00 2001
From: sebastianfis <100689589+sebastianfis@users.noreply.github.com>
Date: Sat, 23 Mar 2024 23:05:56 +0100
Subject: [PATCH] Update edgetpumodel.py

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

diff --git a/edgetpumodel.py b/edgetpumodel.py
index 79e57d1..11e6076 100644
--- a/edgetpumodel.py
+++ b/edgetpumodel.py
@@ -167,7 +167,7 @@ def forward(self, x:np.ndarray, with_nms=True) -> np.ndarray:
         # Scale output
         result = (common.output_tensor(self.interpreter, 0).astype('float32') - self.output_zero) * self.output_scale
         if self.v8:
-            result = np.transpose(result)
+            result = np.transpose(result,[0,2,1])
         
         self.inference_time = time.time() - tstart