Assert non-premature end of JPEG images (#3638)
* premature end of JPEG images * PEP8 reformat Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>pull/3646/head
parent
bb79e13d52
commit
d808855f77
|
@ -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…
Reference in New Issue