Update edgetpumodel.py

sebastianfis · Mar 23, 2024 · 43b9a69 · 43b9a69
1 parent 1f53bca
commit 43b9a69
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/edgetpumodel.py b/edgetpumodel.py
@@ -166,6 +166,9 @@ 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)
+
         self.inference_time = time.time() - tstart
 
         if with_nms: