This commit is contained in:
Glenn Jocher 2021-12-19 16:35:48 +01:00
parent 63f319ac3d
commit 73db0effec

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))
if pred is not None:
print('Predicted:', ' '.join(f'{names[i]:3s}' for i in pred))
print(f'imshow() results saved to {f}')
if __name__ == '__main__':