使用MxNet制作.rec文件時,需要lst文件,所以首先制作lst文件。在incubator-mxnet/tools中的im2rec.py文件可以生成所需要的lst文件。
本文使用cifar10的分類數(shù)據(jù)來制作lst文件,目標檢測所需要的lst文件與分類稍有不同。
-
首先下載cifar10的數(shù)據(jù)集,然后將同一類的圖片放入同一個文件夾,共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

