Browse Source

enhance: imagenet to mindrecord mapfile

tags/v0.7.0-beta
LiuCunwei jonyguo 5 years ago
parent
commit
c133059c59
3 changed files with 13 additions and 13 deletions
  1. +4
    -4
      mindspore/mindrecord/tools/imagenet_to_mr.py
  2. +5
    -5
      model_zoo/utils/cv_to_mindrecord/ImageNet_Similar_Perf/README.md
  3. +4
    -4
      tests/ut/data/mindrecord/testImageNetDataWhole/labels_map.txt

+ 4
- 4
mindspore/mindrecord/tools/imagenet_to_mr.py View File

@@ -35,10 +35,10 @@ class ImageNetToMR:


.. code-block:: .. 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. 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.


+ 5
- 5
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
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 2. Edit run_imagenet.sh and modify the parameters


+ 4
- 4
tests/ut/data/mindrecord/testImageNetDataWhole/labels_map.txt View File

@@ -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

Loading…
Cancel
Save