您的位置:首页 > Web前端

使用caffe的convert_imageset生成lmdb文件

2016-04-16 21:41 288 查看
最近在看caffe的自带的例子时,才发现需要学习的真的很多

其中在将我们的数据转换成lmdb的文件的时候遇到很多问题,现在总结一下,只是个人问题

1.首先在一般是使用图片入手的需要将图片生成键值对的文本文件

在生成的时候可以使用python的脚本或是shell脚本

在次本人写了一个简单的不是很通用的例子,需要的可以去下载,不过要根据自得情况修改。

在后面给出。

在生成完成键值对的文本之后就可以开始进行下一步生成lmdb了

(这里会有一个问题,就是当把键值对的(键,值)的值设置为其他的非数字时在生成lmdb时无法生成lmdb)

也就是生成空数据,这个我也是在无意中发现的。。。

2 开始生成lmdb 这里使用的是caffe自带的函数convert_imageset

在使用convert_imageset 时 可以指定彩色图像和灰度图像(默认的是三通道的彩色图像)

这一点是在转换灰度图像时发现的,会造成训练的时候报错。

灰度图像生成的时候要加上--gray

GLOG_logtostderr=1 $TOOLS/convert_imageset \

--resize_height=$RESIZE_HEIGHT \

--resize_width=$RESIZE_WIDTH \

--shuffle \

--gray \---------------------------------------------------->如果灰度图的话加上即可

$VAL_DATA_ROOT \

$DATA/val.txt \

$EXAMPLE/ilsvrc12_val_lmdb

。在此可以参考example的例子写一个create的shell脚本即可。

3.把生成好的lmdb数据放到对应的prototxt中指定的目录下面即可

在此有时会遇到问题可以参看另一篇博文

就是在进行同一个lmdb在同一个prototxt中多次使用的时需要进行修改路径
http://blog.csdn.net/losteng/article/details/51169965
下面是我的一个文件

有需要的可以免积分参考
http://download.csdn.net/detail/losteng/9493379
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: