你自己的咖啡模型来自第一原则

Your own caffe model from first pricinples

我是咖啡新手。到目前为止,我已经安装了 caffe、pycaffe 和 matcaffe,并查看了一些教程。

我想构建自己的 (lm)db 文件以用于训练模型。

据我所知,caffe.berkeleyvision.com 上的所有教程在创建 lmdb 文件时都使用一组预处理的二进制文件。

我想从基础开始:假设我有一小部分格式正确的图像(jpeg、tiff,无论它是什么)。在这个例子中,这就是我所拥有的。然而,图像的数量很少,所以在这个假设的场景中,我可以自己手动为每张图片分配一个标签。如果你真的要训练一个模型,如果你想了解创建 lmdb 文件的过程,那就不好了,但是 - 黄金!

随意用 lmdb 代替 h5py 或任何最适合情况的东西。

我将如何单独从一组图像创建数据库?

如果您不介意手动标记,则可以创建一个文本文档,其中每个图像都有一行,文件 name/location 以及标签,例如

img1.jpg 0
img2.jpg 1
img3.jpb 3
img4.jpg 0
[...]

然后,要么使用ImageData层,要么使用caffe自带的convert_imageset工具将其转换为lmdb或leveldb。