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训练单通道图像

猜你喜欢

转载自blog.csdn.net/alansss/article/details/107073905