如何改进FCN32的语义分割学习?

时间:2017-02-20 11:55:42

标签: deep-learning caffe pycaffe deeplearning4j

我已经从头开始为我的数据训练FCN32语义分段,我得到了以下输出: enter image description here

可以看出,这不是一个很好的学习曲线,表明对数据的训练不当。 solver如下:

net: "train_val.prototxt"
#test_net: "val.prototxt"
test_iter: 5105 #736
# make test net, but don't invoke it from the solver itself
test_interval: 1000000 #20000
display: 50
average_loss: 50
lr_policy: "step" #"fixed"
stepsize: 50000 #+
gamma: 0.1  #+
# lr for unnormalized softmax
base_lr: 1e-10 
# high momentum
momentum: 0.99
# no gradient accumulation
iter_size: 1
max_iter: 600000
weight_decay: 0.0005
snapshot: 30000
snapshot_prefix: "snapshot/FCN32s_CNN1"
test_initialization: false
solver_mode: GPU

将学习率改为0.001后,情况变得更糟: enter image description here 我想知道如何改进培训?谢谢

1 个答案:

答案 0 :(得分:0)

您可以尝试改变学习率。良好的值通常介于0.1和0.0001之间。