[caffe]将自己的图片数据转换db格式

最新推荐文章于 2020-08-10 14:37:22 发布

jt911405

最新推荐文章于 2020-08-10 14:37:22 发布

阅读量3k

点赞数

分类专栏： DeepLearning 文章标签： deeplearning caffe

本文链接：https://blog.csdn.net/jt911405/article/details/51149107

版权

本文介绍了如何使用Caffe的convert_imageset工具将图片数据转换为适用于训练的LMDB格式。首先，需要准备4个文件，包括训练和测试数据以及对应的txt标签文件。接着，创建一个sh脚本，指定数据路径、图片尺寸和输出LMDB文件的位置，执行脚本以生成训练和测试的LMDB文件。在过程中可能会遇到的问题，如需要确保脚本完整并正确执行每个步骤。

摘要由CSDN通过智能技术生成

我们手中有的一般都是图片数据，jpg、bmp格式等，但caffe常使用的数据是db格式（leveldb/lmdb），因此首先我们要将自己数据转换成caffe可运行的格式文件。别捉鸡，caffe有给我们提供工具。根目录的tools文件下convert_imageset.cpp，经编译生成可执行文件，存放于./build/tools/文件，这个文件可帮我们将图片文件转换成db格式。

以分类任务为例，需要4个文件，train、train.txt、val、val.txt，train和val分别存放训练数据和测试数据，而两个txt文件分别是其文件的lable。

我们可以创建一个sh脚本来执行所有的操作。（以下操作均在/path to/caffe-master/路径下）

sudo vim data/mydata/create_lmdb.sh（我的数据放在data文件下的mydata目录下）

创建好脚本文件后即在里面写入代码：

#！ /usr/bin/env sh

DATA = data/m