Update edgetpumodel.py

sebastianfis · Mar 23, 2024 · b59317a · b59317a
1 parent 0a2529e
commit b59317a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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