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

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.

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

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