MxNet 制作rec文件
2018-02-07 本文已影响0人
Lxiaoming
使用MxNet制作.rec文件时,需要lst文件,所以首先制作lst文件。在incubator-mxnet/tools
中的im2rec.py
文件可以生成所需要的lst文件。
本文使用cifar10的分类数据来制作lst文件,目标检测所需要的lst文件与分类稍有不同。
-
首先下载cifar10的数据集,然后将同一类的图片放入同一个文件夹,共10类,即10个文件夹。
10个文件夹
- 制作lst文件,即生成train.lst文件
python im2rec.py train ~/Data/CIFAR-10/train_valid_test/train/ --list --recursive
-
train.lst文件格式
train.lst
- 生成train.rec文件
python im2rec.py train ~/Data/CIFAR-10/train_valid_test/train/ --num-thread=4