提交 c133059c 编写于 作者: L LiuCunwei 提交者: jonyguo

enhance: imagenet to mindrecord mapfile

上级 b55e5e2c
......@@ -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
......
n00000005 0 data_line
n00000006 1 small_iron_box
n00000007 2 plastic_toothpicks
n00000002 3 orange
n00000005 0
n00000006 1
n00000007 2
n00000002 3
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册