Windows+caffe对自己的样本进行特征提取

xiaoxiao2021-02-28  105

生成lmdb/leveldb文件+均值文件

利用caffe对自己的图片生成leveldb格式文件: http://www.cnblogs.com/TensorSense/p/6744075.html 1、将图片进行分类,建立train和test文件夹存放训练和测试的图像 2、建立txt标签文件存放图片路径以及类别 其中,需要注意的是,——在文件名定义的时候标签与文件名之间只能有一个空格,多了会产生错误。 ——在使用路径时,注意用绝对路径还是相对路径 3、新建.bat文件调用convert_imageset.exe转化图片格式: D:\caffe\caffe-master\Build\x64\Release\convert_imageset.exe --resize_width=64 --resize_height=64 ./sub1/ train.txt train_leveldb -backend=leveldb D:\caffe\caffe-master\Build\x64\Release\convert_imageset.exe --resize_width=28 --resize_height=28 ./sub2/ test.txt test_leveldb -backend=leveldb Pause

这里的每行包含七个部分: 第一个部分:D:\caffe-master\Build\x64\Release\convert_imageset.exe 表示 convert_imageset.exe所在路径(PS,若是用Debug生成的,则在将Release改成Debug即可) 第二、三部分,是对原始图片进行了resize 第四部分: ./train/ 是训练图片文件所在路径,此处用了相对路径的形式(不了解相对路径的同学,请度娘 ./ ../的作用) 第五部分,train_labels.txt是训练集的标签 第六部分:train_lmdb 是生成LMDB格式文件的文件名 第七部分:-backend=lmdb 是所要转成的格式(要转成 leveldb的则改成 -backend=leveldb) 然后双击运行

运行正常的话,在文件夹下会出现 test_leveldb 和train_leveldb 两个文件夹。至此,已经将自己的数据集转换成为leveldb格式的数据了,即caffe可读的数据格式。(将文件中leveldb换为lmdb即可更改格式) 双击.bat文件运行之后生成文件:

对生成的train_lmdb进行均值文件获取: SET GLOG_logtostderr=1 compute_image_mean.exe train_lmdb image_mean.binaryproto pause 我是直接在cmd中运行的,生成了image_mean.binaryproto 文件

对生成文件进行特征提取

转载请注明原文地址: https://www.6miu.com/read-50415.html

最新回复(0)