forked from OSSInnovation/mindspore
enhance: imagenet to mindrecord mapfile
This commit is contained in:
parent
b55e5e2ce2
commit
c133059c59
|
@ -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.
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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…
Reference in New Issue