diff --git a/demo/imagenet_reader.py b/demo/imagenet_reader.py index 01aa65a50ea4d556a89beff0541b9ddae87be604..947fd023b9c6eea5f4d6d0a5d52337b1ba97cc3f 100644 --- a/demo/imagenet_reader.py +++ b/demo/imagenet_reader.py @@ -100,7 +100,11 @@ def distort_color(img): def process_image(sample, mode, color_jitter, rotate): img_path = sample[0] - img = Image.open(img_path) + try: + img = Image.open(img_path) + except: + print(img_path, "not exists!") + return None if mode == 'train': if rotate: img = rotate_image(img) img = random_crop(img, DATA_DIM)