Add `print(results)` override for PyTorch Hub results (#7559)

2022-04-24 12:07:05 -07:00 · 2022-04-24 12:07:05 -07:00 · c16671fc74
parent be67572279
commit c16671fc74
1 changed files with 5 additions and 1 deletions
--- a/models/common.py
+++ b/models/common.py
@ -683,7 +683,11 @@ class Detections:
        return x

    def __len__(self):
-        return self.n
+        return self.n  # override len(results)
+
+    def __str__(self):
+        self.print()  # override print(results)
+        return ''


 class Classify(nn.Module):