ultralytics 8.3.26 EdgeTPU Pose models fix (#17281)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-10-31 20:36:27 +08:00 · 2024-10-31 20:36:27 +08:00 · f4e7756bff
commit f4e7756bff
parent e8743f2ac9
3 changed files with 19 additions and 4 deletions
--- a/ultralytics/nn/autobackend.py
+++ b/ultralytics/nn/autobackend.py
@ -663,6 +663,9 @@ class AutoBackend(nn.Module):
                        else:
                            x[:, [0, 2]] *= w
                            x[:, [1, 3]] *= h
+                            if self.task == "pose":
+                                x[:, 5::3] *= w
+                                x[:, 6::3] *= h
                    y.append(x)
            # TF segment fixes: export is reversed vs ONNX export and protos are transposed
            if len(y) == 2:  # segment with (det, proto) output order reversed