新版本caffe不需要lmdb或者leveldb了,咋训练

Caffe1也可以直接读图像数据,用ImageData层实现,可以看链接:Caffe | ImageData Layer。实现起来也很简单,除了需要图像数据,还需要label列表。Imdb格式好处是稳定,但是缺点是占内存,而且每次更新数据集的话就要重新生成新的Imdb文件,比较麻烦。

■网友
用ImageData层的例子,需要注意的是训练的图像文件名不能有空格,还有Label.txt的文件名(可以带目录)和类别标签之间用空格隔开,不能用tab,我看源代码生成levelDB是空格和tab都可以的,用ImageData只能用空格隔开
【新版本caffe不需要lmdb或者leveldb了,咋训练】 layer {
top: "data"
top: "label"
name: "data"
type: "ImageData"
image_data_param {
source: "D:/Project/Recognition/Label.txt"
root_folder: "D:/Project/Recognition/"
batch_size: 32
shuffle: true #每个epoch都会进行shuffle
new_height: 66
new_width: 519
}
include {
phase: TRAIN
}
}


    推荐阅读