关于Segnet训练过程中的几个问题

环境

操作系统: ubuntu14.04
caffe: caffe-segnet(官方)
model: segnet-Tutorial

需要注意的问题

  1. 比较segnet_model_driving_webdemo.prototxt和segnet_inference.prototxt发现,只有卷积的权重初始化方式(w和b)和最后一层不同(webdemo为’argmax’而前者为’softmax’).
  2. 由于官方的segnet model zoo只提供了部分训练好的权重,并且一些权重存在问题(例如segnet_basic_camvid.caffemodel只有5.7M,以及bayesian_segnet_basic_camvid.caffemodel也只有6M),而论文中说segnet的模型大小是117M,说明caffemodel上传的应该是错误的.结合问题1,暂时就使用webdemo的caffemodel进行测试.
  3. segnet和webdemo的识别类别是不一样的,前者是11,后者是12;别切不同类别的所使用的颜色表示也是不一样的.所以需要如果使用webdemo的caffemodel对test_segmentatino_camvid.py进行预测,需要添加预测部分的颜色.可以参考https://github.com/alexgkendall/SegNet-Tutorial/issues/51.
  4. 关于如果使用webdemo进行预测,还需不需要进行batch normalization.测试了一次发现,不能使用batch normalization,如果使用bn对webdemo.caffemodel进行处理,结果变得很差.以下分别是不使用bn和使用bn获得的结果.
    1
    2
  5. 自己使用cityscapes的数据集进行了训练,在vgg16上进行finetune,使用cityscapes-fine数据,resize到480x240.训练了将近100轮,相关结果之后会记录.
  • 2
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值