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

!3814 enhance: map file label start begin 0 in imagenet_to_mr tool 

Merge pull request !3814 from liucunwei/enhance_map_file_for_imagenet
This commit is contained in:
mindspore-ci-bot 2020-08-01 15:01:45 +08:00 committed by Gitee
parent 11f786c9bd c133059c59
commit 5a3b557354
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::
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.

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

8
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