From d7c94c31a0f50f23a03a94999be2bbe850bf6e62 Mon Sep 17 00:00:00 2001
From: unknown <xiaowk5516@163.com>
Date: Wed, 16 Jun 2021 18:44:04 +0800
Subject: [PATCH] premature end of JPEG images

---
 utils/datasets.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/utils/datasets.py b/utils/datasets.py
index 20109e739c02..fb784517c85f 100755
--- a/utils/datasets.py
+++ b/utils/datasets.py
@@ -1057,6 +1057,11 @@ 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