diff --git a/mindspore/mindrecord/tools/imagenet_to_mr.py b/mindspore/mindrecord/tools/imagenet_to_mr.py index 8c8de689c1fc271afa4eedfb58e5b4ea76f81b71..e941e764774b1c7cc23b442ad7a031e50f135472 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