Browse Source

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

* premature end of JPEG images

* PEP8 reformat

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
modifyDataloader
xiaowk5516 GitHub 3 years ago
parent
commit
d808855f77
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      utils/datasets.py

+ 4
- 0
utils/datasets.py View File

@@ -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

Loading…
Cancel
Save