update

4 weeks ago · 4182ddb63d
parent f9bc0e0a46
commit 4182ddb63d
1 changed files with 1 additions and 3 deletions
--- a/ultralytics/nn/autobackend.py
+++ b/ultralytics/nn/autobackend.py
@ -555,9 +555,7 @@ class AutoBackend(nn.Module):
                self.session.run_with_iobinding(self.io)
                y = self.bindings
            if self.mct:
-                from ultralytics.utils.ops import xyxy2xywh
+                y = np.concatenate([y[0], y[1][:, :, None], y[2][:, :, None]], axis=-1)
                y = np.concatenate([xyxy2xywh(y[0]), y[1]], axis=-1).transpose(0, 2, 1)
        # OpenVINO
        elif self.xml: