使用caffe的convert_imageset生成lmdb文件

本文介绍了使用 Caffe 框架时数据预处理的方法,包括如何生成键值对文本文件,解决非数字标签问题,以及如何使用 convert_imageset 函数生成 LMDB 数据库,并针对灰度图像进行特别说明。

最近在看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.youkuaiyun.com/losteng/article/details/51169965

下面是我的一个文件

有需要的可以免积分参考

http://download.youkuaiyun.com/detail/losteng/9493379


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值