您的位置:首页 > Web前端

采用caffe的model在自己的数据上面进行 fine-tuning

2014-07-23 10:59 232 查看
学了caffe的基本构架以后,想采用caffe的model在自己的数据库上面进行fine-tuning.

我的主要工作还是做图像分类,所以我没有修改caffe的构架。

(1)只是将最后的输出1000改为了14,同时把train和val中的fc8全部改为一个自定义的名字,如fc-ft,这个名字随

便自己定义。

(2)用自带的create_imagenet.sh脚本生成训练和验证数据,用make_imagenet_mean.sh生成均值文件;

(3)编写运行脚本,finetune.sh:并且注意自己的相关文件的路径。

#!/usr/bin/env sh

TOOLS=../../build/tools

GLOG_logtostderr=1 $TOOLS/finetune_net.bin std_solver.prototxt caffe_reference_imagenet_model

echo "Done."
(4)./finetune.sh,即可。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐