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

    技术2022-07-11  79

    主要就是两种方法, (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训练单通道图像

    Processed: 0.014, SQL: 9