Caffe入门学习(7)将自己的图像数据转成lmdb格式

对模型训练的第一步来说,就是进行格式转换,将常用的图片(jpg、png)转换为caffe使用的lmdb或levdb数据库格式,由于leveldb是Caffe早期的数据格式,现在主要都是用lmdb了。

目标:
- 将训练和测试数据转化成lmdb数据库

工具:
caffe中提供了将图片转化为lmdb的可执行文件
./caffe/Build/x64/Release/convert_imageset

convert_imageset [FLAGS] ROOTFOLDER/ LISTFILE DB_NAME

需要带四个参数:
FLAGS: 图片参数组,后面详细介绍
ROOTFOLDER: 图片存放的绝对路径,从linux系统根目录开始
LISTFILE: 图片文件列表清单,一般为一个txt文件,一行一张图片
DB_NAME: 最终生成的db文件存放目录

下面是具体转换步骤
:当前我的学习环境都是在windows下

1、建立目录

./caffe/data/myemaple下建立目录结构

train
  0 -- 图片0
  1 -- 图片1
test
  图片0+图片1 混合

2、取得图片列表

打开命令行工具,在目录下执行

dir/s/on/b >train.txt 

就可以生成
这里写图片描述

调整路径并加上分类标签,就是这样:
这里写图片描述

3、创建脚本

比如叫 createlmdb.bat
这里仍然是借助于Caffe生成的工具convert_imageset

 E:/caffe/Build/x64/Release/convert_imageset --shuffle ^
 --resize_height=256 --resize_width=256 ^
 E:/caffe/data/csdb/train/ ^
 E:/caffe/data/csdb/train.txt ^
 E:/caffe/data/csdb/train_lmdb ^
 -backend=lmdb

pause

设置参数-shuffle,打乱图片顺序。
设置参数-resize_height和-resize_width将所有图片尺寸都变为256*256.

4、转换格式

双击脚本执行
这里写图片描述

生成结果
这里写图片描述

检查一下文件大小,只要不是太小或为空,一般就对了。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章