diff --git a/mindspore/mindrecord/tools/imagenet_to_mr.py b/mindspore/mindrecord/tools/imagenet_to_mr.py index 0aa870384e6..59695c87340 100644 --- a/mindspore/mindrecord/tools/imagenet_to_mr.py +++ b/mindspore/mindrecord/tools/imagenet_to_mr.py @@ -35,10 +35,10 @@ class ImageNetToMR: .. code-block:: - n02119789 1 pen - n02100735 2 notebook - n02110185 3 mouse - n02096294 4 orange + n02119789 0 + n02100735 1 + n02110185 2 + n02096294 3 image_dir (str): image directory contains n02119789, n02100735, n02110185, n02096294 dir. destination (str): the MindRecord file path to transform into. diff --git a/model_zoo/utils/cv_to_mindrecord/ImageNet_Similar_Perf/README.md b/model_zoo/utils/cv_to_mindrecord/ImageNet_Similar_Perf/README.md index 8bdcb9e25df..d3bd5fdc182 100644 --- a/model_zoo/utils/cv_to_mindrecord/ImageNet_Similar_Perf/README.md +++ b/model_zoo/utils/cv_to_mindrecord/ImageNet_Similar_Perf/README.md @@ -28,12 +28,12 @@ This example provides an efficient way to generate MindRecord. Users only need t Store the downloaded ImageNet dataset in a folder. The folder contains all images and a mapping file that records labels of the images. - In the mapping file, there are three columns, which are separated by spaces. They indicate image classes, label IDs, and label names. The following is an example of the mapping file: + In the mapping file, there are three columns, which are separated by spaces. They indicate image classes and label IDs. The following is an example of the mapping file: ``` - n02119789 1 pen - n02100735 2 notbook - n02110185 3 mouse - n02096294 4 orange + n02119789 0 + n02100735 1 + n02110185 2 + n02096294 3 ``` 2. Edit run_imagenet.sh and modify the parameters diff --git a/tests/ut/data/mindrecord/testImageNetDataWhole/labels_map.txt b/tests/ut/data/mindrecord/testImageNetDataWhole/labels_map.txt index 73e9759b899..82082e1c747 100644 --- a/tests/ut/data/mindrecord/testImageNetDataWhole/labels_map.txt +++ b/tests/ut/data/mindrecord/testImageNetDataWhole/labels_map.txt @@ -1,4 +1,4 @@ -n00000005 0 data_line -n00000006 1 small_iron_box -n00000007 2 plastic_toothpicks -n00000002 3 orange +n00000005 0 +n00000006 1 +n00000007 2 +n00000002 3