参考了一些其他人的博客,有一些不一样的地方,记录一下
主要分两步:生成.lst和生成.rec
1.1、生成.lst
需要准备的就是你的图像。~/image是你放图片的路径,image下可以有子文件夹,子文件夹里面放图片,也可以把图片直接放到image路径下。~/lst是存放生成的.lst的文件夹,data是生成的.lst的名称前缀,mxnet的路径是~/anaconda3/lib/python3.7/site-packages/mxnet,那么运行下面的命令就可以生成.lst文件:
python ~/anaconda3/lib/python3.7/site-packages/mxnet/tools/im2rec.py --list --recursive ~/lst/data ~/image
1.2、生成.rec
需要准备的就是第一步生成的.lst文件和你的图像。
假设你要将生成的.rec文件放在.lst文件相同的/home/lst文件夹下(一般都会这样操作),那么运行下面的命令就可以生成.rec文件:
python ~/anaconda3/lib/python3.7/site-packages/mxnet/tools/im2rec.py --num-thread 4 ~/lst ~/image