diff --git a/enhance.py b/enhance.py index becb851..8331506 100755 --- a/enhance.py +++ b/enhance.py @@ -170,6 +170,7 @@ def add_to_buffer(self, f): if any(s < args.batch_shape for s in orig.size): raise ValueError('Image is too small for training with size {}'.format(orig.size)) except Exception as e: + print(e) warn('Could not load `{}` as image.'.format(filename), ' - Try fixing or removing the file before next run.') self.files.remove(f)