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

ctjanuhowski · Sep 8, 2022 · 42e3b6f · 42e3b6f
1 parent db47205
commit 42e3b6f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/models/common.py b/models/common.py
@@ -683,7 +683,11 @@ def tolist(self):
         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):