diff --git a/utils/datasets.py b/utils/datasets.py index eea8ad3..852bb7c 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -873,7 +873,7 @@ def verify_image_label(args): with open(im_file, 'rb') as f: f.seek(-2, 2) if f.read() != b'\xff\xd9': # corrupt JPEG - im.save(im_file, format='JPEG', subsampling=0, quality=100) # re-save image + Image.open(im_file).save(im_file, format='JPEG', subsampling=0, quality=100) # re-save image msg = f'{prefix}WARNING: corrupt JPEG restored and saved {im_file}' # verify labels