mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

Update

Browse Source
This commit is contained in:
Glenn Jocher 2021-12-19 16:35:48 +01:00
parent 63f319ac3d
commit 73db0effec
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
classifier.py
View File

@ -242,6 +242,7 @@ def imshow(img, labels=None, pred=None, names=None, f=Path('images.jpg')):
print('True: ', ' '.join(f'{names[i]:3s}' for i in labels)) print('True: ', ' '.join(f'{names[i]:3s}' for i in labels))
if pred is not None: if pred is not None:
print('Predicted:', ' '.join(f'{names[i]:3s}' for i in pred)) print('Predicted:', ' '.join(f'{names[i]:3s}' for i in pred))
print(f'imshow() results saved to {f}')
if __name__ == '__main__': if __name__ == '__main__':