您自己的 caffe 模型来自第一原则

Question

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

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

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

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

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

我将如何仅从一组图像创建数据库？

Answer 1

如果您对手工标记没问题，您可以创建一个文本文档，其中每个图像都有一行，其中包含文件名/位置以及标签，例如

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

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