深度学习-采用caffe提供的工具extract_features.bin进行特征提取

地1,在caffe目录下建立一临时目录_temp
mkdir examples/_temp
2,生成一个特提取图像的文件名列表文件
find `pwd`/examples/images/myimages/ -type f -exec echo {} \; > examples/_temp/temp.txt

3,每一个图像文件名的后面都期望有一个分类标签,因此,每一行路径的后面加一个0代表结束

sed "s/$/ 0/" examples/_temp/temp.txt > examples/_temp/file_list.txt

在下面的配置文件中要用到


4,使用extract_features.bin工具提取特征
./build/tools/extract_features.bin examples/testFinetune/my_mode_iter_5000.caffemodel.h5 examples/testFinetune/deploy.prototxt fc7 examples/_temp/features 50 leveldb

模型:my_mode_iter_5000.caffemodel.h5
网络结构:deploy.prototxt
层的名称:fc7,也可以是其他层
特征存储的位置:examples/_temp/features
个数:50
特征存储的文件格式:leveldb,这里面采用leveldb数据库的形式

注意:这里面用的模型、网络结构都是事先前训练好的。主

./build/tools/extract_features.bin examples/testFinetune/my_mode_iter_5000.caffemodel.h5 examples/testFinetune/deploy.prototxt fc7 examples/_temp/features 50 leveldb GPU

提醒:最后一个参数要是使用gpu的话,参数必须大写GPU。读代码的结果,参考资料太少了,读代码还是很爽!


参考资料:http://caffe.berkeleyvision.org/gathered/examples/feature_extraction.html。官方文档

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值