如何在Deep Lab V2中部署模型

时间:2016-08-08 23:00:28

标签: caffe

我正在尝试训练和测试Deep CNN模型以进行细分。我正在使用基于Caffe的Deep Lab V2模型。我已经成功完成了run_pascal.sh脚本的教程来训练模型。但是,如果我想训练自己的分段模型并进行部署,我该如何编写deploy.prototxt文件?任何人都可以根据最新的V2版本提供部署文件的示例吗?

1 个答案:

答案 0 :(得分:0)

我曾经使用CMD训练模型:

caffe.bin train --solver = solver.prototxt

我的solver.prototxt如下:

train_net:" train.prototxt"

lr_policy:" poly"

力量:0.9

base_lr:5e-2

average_loss:1

显示:20

max_iter:10000

weight_decay:0.0005

快照:10000

snapshot_prefix:" / path / to / save / snapshots"

输入:" AdaDelta"

solver_mode:GPU

train.prototxt是用于训练的模型

确保在必要时包含路径