fix: change field name from 'data' to 'image' in mindrecord which similar with ImageFolderDatasetV2

example/cv_to_mindrecord/ImageNet_Similar_Perf/imagenet/mr_api.py

@@ -26,8 +26,8 @@ import os
 import pickle
 
 ######## mindrecord_schema begin ##########
-mindrecord_schema = {"label": {"type": "int64"},
+mindrecord_schema = {"label": {"type": "int32"},
-                     "data": {"type": "bytes"},
+                     "image": {"type": "bytes"},
                      "file_name": {"type": "string"}}
 ######## mindrecord_schema end ##########
 
@@ -121,5 +121,5 @@ def mindrecord_dict_data(task_id):
         if not image_bytes:
             print("The image file: {} is invalid.".format(file_name))
             continue
-        data["data"] = image_bytes
+        data["image"] = image_bytes
         yield data

mindspore/mindrecord/tools/imagenet_to_mr.py

@@ -114,7 +114,7 @@ class ImageNetToMR:
                 if not image_bytes:
                     logger.warning("The image file: {} is invalid.".format(file_name))
                     continue
-                data["data"] = image_bytes
+                data["image"] = image_bytes
                 yield data
 
     def transform(self):
@@ -126,8 +126,8 @@ class ImageNetToMR:
         """
         t0_total = time.time()
 
-        imagenet_schema_json = {"label": {"type": "int64"},
+        imagenet_schema_json = {"label": {"type": "int32"},
-                                "data": {"type": "bytes"},
+                                "image": {"type": "bytes"},
                                 "file_name": {"type": "string"}}
 
         logger.info("transformed MindRecord schema is: {}".format(imagenet_schema_json))