caffe_ssd-matlab调试

本文详细介绍了SSD_FACE模型的训练配置文件设置,包括solver.prototxt中的参数配置如学习率、迭代次数等,以及train.prototxt中各层的具体设置。特别强调了根据类别数量调整输出的重要性。

models:
-solver.prototxt: 各种参数(包括train.prototxt的路径)

train_net: “models/Context_Net/train.prototxt”
base_lr: 0.0005
display: 10
max_iter: 60000
lr_policy: “step”
gamma: 0.1
momentum: 0.9
weight_decay: 0.0005
stepsize: 40000
snapshot: 40000
snapshot_prefix: “models/SSD_FACE_300x300”
solver_mode: GPU
device_id: 0
debug_info: false
snapshot_after_train: true
test_initialization: false
average_loss: 10
iter_size: 2
type: “SGD”
eval_type: “detection”
ap_version: “11point”

-train.prototxt: 训练各层设置

需要修改最底层的conv4_3…conf的output的个数为6*类数
最底层的其他…conf的output的个数为3*类数

-VGG_ILSVRC_16_layers_fc_reduced.caffemodel

functions
-+Net/net_conf.m

修改num_classes类数,数据路径

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值