制作带多标签的.rec作为MXNET的输入

使用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然后进行训练


猜你喜欢

转载自blog.csdn.net/baidu_26408419/article/details/79179020