From b4151efe81ffd7ad9a345d7cf9cb56621d18f8f1 Mon Sep 17 00:00:00 2001
From: jonyguo <guozhijian@huawei.com>
Date: Fri, 24 Apr 2020 17:27:14 +0800
Subject: [PATCH] fix: imagenet to mindrecord failed when the pic size is 0

---
 mindspore/mindrecord/tools/imagenet_to_mr.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/mindspore/mindrecord/tools/imagenet_to_mr.py b/mindspore/mindrecord/tools/imagenet_to_mr.py
index 8c8de689c1..e941e76477 100644
--- a/mindspore/mindrecord/tools/imagenet_to_mr.py
+++ b/mindspore/mindrecord/tools/imagenet_to_mr.py
@@ -111,6 +111,9 @@ class ImageNetToMR:
                 image_file = open(file_name, "rb")
                 image_bytes = image_file.read()
                 image_file.close()
+                if not image_bytes:
+                    logger.warning("The image file: {} is invalid.".format(file_name))
+                    continue
                 data["data"] = image_bytes
                 yield data