OSSInnovation
/
mindspore
1
0
Fork 26
Code Issues Pull Requests 7 Releases Wiki Activity

enhance: imagenet to mindrecord mapfile

This commit is contained in:
LiuCunwei 2020-07-31 15:26:42 +08:00 committed by jonyguo
parent b55e5e2ce2
commit c133059c59
3 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/mindrecord/tools/imagenet_to_mr.py
View File

@ -35,10 +35,10 @@ class ImageNetToMR:
.. code-block:: .. code-block::
n02119789 1 pen n02119789 0
n02100735 2 notebook n02100735 1
n02110185 3 mouse n02110185 2
n02096294 4 orange n02096294 3
image_dir (str): image directory contains n02119789, n02100735, n02110185, n02096294 dir. image_dir (str): image directory contains n02119789, n02100735, n02110185, n02096294 dir.
destination (str): the MindRecord file path to transform into. destination (str): the MindRecord file path to transform into.

10
model_zoo/utils/cv_to_mindrecord/ImageNet_Similar_Perf/README.md
View File

@ -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. 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 n02119789 0
n02100735 2 notbook n02100735 1
n02110185 3 mouse n02110185 2
n02096294 4 orange n02096294 3
``` ```
2. Edit run_imagenet.sh and modify the parameters 2. Edit run_imagenet.sh and modify the parameters

8
tests/ut/data/mindrecord/testImageNetDataWhole/labels_map.txt
View File

@ -1,4 +1,4 @@
n00000005 0 data_line n00000005 0
n00000006 1 small_iron_box n00000006 1
n00000007 2 plastic_toothpicks n00000007 2
n00000002 3 orange n00000002 3