Unverified 提交 d808855f authored 作者: xiaowk5516's avatar xiaowk5516 提交者: GitHub

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

* premature end of JPEG images * PEP8 reformat Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
上级 bb79e13d
...@@ -1057,6 +1057,10 @@ def verify_image_label(args): ...@@ -1057,6 +1057,10 @@ def verify_image_label(args):
shape = exif_size(im) # image size shape = exif_size(im) # image size
assert (shape[0] > 9) & (shape[1] > 9), f'image size {shape} <10 pixels' 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}' 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 # verify labels
segments = [] # instance segments segments = [] # instance segments
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论