caffe 分类任务中训练与测试网络命令格式

来源:互联网 发布:mac 搜狗输入法设置 编辑:程序博客网 时间:2024/05/22 03:08

1、训练网络是的指令(该命令为迁移已有的Alexnet权重,如若从头训练则可将--weight字段去掉即可)

TOOLS=./build/tools                                                                                                    

  

   $TOOLS/caffe train --solver models/llp/alex_decon2_fc/solver.prototxt --weights /home/k40/caffe-master/models/bvlc_alexnet/bvlc_alexnet.caffemodel --gpu 0 2>&1 | tee ./models/llp/alex_decon2_fc/log.txt


2、测试网络的分类准确率

./build/tools/caffe test --model=/caffe-master/models/llp/alex_decon2_fc/train_val.prototxt --weights=/caffe-master/models/llp/alex_decon2_fc/caffenet_train_iter_50000.caffemodel --gpu=0