Unverified 提交 8fcbe434 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Check im.format during dataset caching (#2042)

* Check im.format during dataset caching * Update datasets.py
上级 a41d910c
......@@ -445,7 +445,8 @@ class LoadImagesAndLabels(Dataset): # for training/testing
im = Image.open(im_file)
im.verify() # PIL verify
shape = exif_size(im) # image size
assert (shape[0] > 9) & (shape[1] > 9), 'image size <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}'
# verify labels
if os.path.isfile(lb_file):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论