diff --git a/classifier.py b/classifier.py index 5811aeee6..1c12c1a06 100644 --- a/classifier.py +++ b/classifier.py @@ -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__':