使用MXNET时,第一步就是制作符合mxnet要求的输入格式。
当我们的任务是:x是图像,然后y是多(或单)标签的输出时,我们可以制作符合mxnet要求的.rec格式:
具体详见mxnet教程:https://mxnet.incubator.apache.org/api/scala/io.html
英文官网流程:
中文注释:
①生成.lst文件,格式:
Index label_0 label_1 label_2 label_3path
path:图片名字。
②生成.rec文件(找到toolim2rec 运行文件,配置MXNET时候要用make方式编译才有这个工具)
命令行输入:.../bin/im2rec a.lst 图片路径/
a.rec label_width=4
③mx.io.ImageRecordIter加载.rec然后进行训练
热门工具 换一换