caffe-ssd采用单通道的灰度图像进行训练

主要就是两种方法,
(1) 一种就是伪单通道
在transform_param{…}中添加force_color:True,将灰度图像强制转换为彩色图像进行训练’
(2) 修改网络为单通道输入
(2.1)create_data.sh中create_annoset.py的输入参数添加–gray,生成单通道lmdb格式的数据
(2.2)train.prototxt中的数据层 AnnotatedData 层中的mean_value只保留一个通道的,同时将和颜色通道数据增强相关的参数去掉(distort_param{…})
(2.3)test.prototxt中的mean_value也只保留一个通道的
(2.4)deploy.prototxt中的input_shape中的dim由3改为1
然后就可以进行单通道图像训练了。相应的,在测试时也需要将读入的jpg格式图像提取单通道后再进行测试。

参考:
1.github上ssd源码作者答疑:How to use SSD to train and test images with only one channel?
2.将灰度图转彩色图的方法:SSD训练单通道图像

  • 1
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值