From d7c94c31a0f50f23a03a94999be2bbe850bf6e62 Mon Sep 17 00:00:00 2001 From: unknown Date: Wed, 16 Jun 2021 18:44:04 +0800 Subject: [PATCH 1/2] premature end of JPEG images --- utils/datasets.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/utils/datasets.py b/utils/datasets.py index 20109e739c02..fb784517c85f 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -1057,6 +1057,11 @@ def verify_image_label(args): shape = exif_size(im) # image size assert (shape[0] > 9) & (shape[1] > 9), f'image size {shape} <10 pixels' assert im.format.lower() in img_formats, f'invalid image format {im.format}' + if im.format.lower() in ('jpg', 'jpeg'): + with open(im_file, 'rb') as f: + f.seek(-2, 2) + assert f.read() == b'\xff\xd9', 'corrupted JPEG' + # verify labels segments = [] # instance segments From 1f81cdb85f6f3a02fd2768239f22f235a85dc064 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 16 Jun 2021 13:19:18 +0200 Subject: [PATCH 2/2] PEP8 reformat --- utils/datasets.py | 1 - 1 file changed, 1 deletion(-) diff --git a/utils/datasets.py b/utils/datasets.py index fb784517c85f..a1a8fa8f32a9 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -1062,7 +1062,6 @@ def verify_image_label(args): f.seek(-2, 2) assert f.read() == b'\xff\xd9', 'corrupted JPEG' - # verify labels segments = [] # instance segments if os.path.isfile(lb_file):