使MxNet的im2rec.py生成rec文件

一,安装MxNet

使用命令

pip2 install mxnet-cu80

安装基于CUDA8.0的GPU版本MxNet,版本为:1.2.1,安装目录为:”/usr/local/lib/python2.7/dist-packages/mxnet“

二,制作.lst文件

使用命令

python2  /usr/local/lib/python2.7/dist-packages/mxnet/tools/im2rec.py  --list  --recursive test  /home/open/Data/CIFAR/test

或者

python2  /usr/local/lib/python2.7/dist-packages/mxnet/tools/im2rec.py  --list --recursive --no-shuffle test  /home/open/Data/CIFAR/test  #生成不是乱序的

test为生成的.lst文件的文件名,最后的路径为原始图像所在的目录。

注意,生成的.lst文件的文件名必须要和原始图像所子啊的文件夹的文件名相一致,否则下面生成.rec文件会找不到图像。

三,制作.rec文件

有了.lst文件,使用下面的命令制作.rec文件

python2  /usr/local/lib/python2.7/dist-packages/mxnet/tools/im2rec.py  --pass-through  test.lst  test

或者

python2  /usr/local/lib/python2.7/dist-packages/mxnet/tools/im2rec.py  --pass-through  -num-thread 4  test.lst  test  #多线程

 

 

你可能感兴趣的:(使MxNet的im2rec.py生成rec文件)