Assert non-premature end of JPEG images (ultralytics#3638)

* premature end of JPEG images * PEP8 reformat Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
fhkiel-mlaip · Jun 16, 2021 · 115d77b · 115d77b
1 parent 0319d63
commit 115d77b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/utils/datasets.py b/utils/datasets.py
@@ -1057,6 +1057,10 @@ 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