mxnet | 制作rec数据文件-2. 生成 .rec

需要准备的是第一步生成的 .lst 文件和你的图像
假设将要生成的 .rec 文件放在 .lst 文件相同的 /home/lst 文件夹下 (一般都会这样操作),那么运行下面的命令就可以生成 .rec 文件:

python ~/incubator-mxnet/tools/im2rec.py --num-thread 4 /home/lst /home/image

这里倒数第二个参数:/home/lst是你的.lst文件所放的路径,可以不用指明.lst文件名称,因为代码会自动搜索/home/lst文件夹下所有以.lst结尾的文件。最后一个参数:/home/image 是你的图像所放的路径。--num-thread 4 这个参数是表示用4个线程来执行,当你数据量较大的时候,生成 .rec 的过程会比较慢,这样可以加速

运行成功后,在 /home/rec 文件夹下就成成了 data_train.rec 和 data_val.rec 文件,然后就可以用 mxnet.io.ImageRecordlter 类来导入 .rec 文件了。

另外还会生成两个 .idx 文件,可忽略。

上一篇:算法学习——LeetCode力扣动态规划篇2(343. 整数拆分、96. 不同的二叉搜索树、416. 分割等和子集、1049. 最后一块石头的重量 II)


下一篇:【uniapp】Vue3移动端滚动加载 分页组件的封装