Add `print(results)` override for PyTorch Hub results (#7559)
Browse files- models/common.py +5 -1
models/common.py
CHANGED
@@ -683,7 +683,11 @@ class Detections:
|
|
683 |
return x
|
684 |
|
685 |
def __len__(self):
|
686 |
-
return self.n
|
|
|
|
|
|
|
|
|
687 |
|
688 |
|
689 |
class Classify(nn.Module):
|
|
|
683 |
return x
|
684 |
|
685 |
def __len__(self):
|
686 |
+
return self.n # override len(results)
|
687 |
+
|
688 |
+
def __str__(self):
|
689 |
+
self.print() # override print(results)
|
690 |
+
return ''
|
691 |
|
692 |
|
693 |
class Classify(nn.Module):
|