如何在Caffe旧版IMAGE_DATA层将图像转为灰度图?
解决方案:旧版IMAGE_DATA层设置单通道灰度图读取
当然可以实现!旧版的IMAGE_DATA层确实有专门的参数来控制图像是否以灰度模式读取,从而将通道数保持为1。
你只需要在image_data_param配置块中添加is_color: false参数即可——这个参数默认值为true(即默认读取3通道彩色图像),将其设为false后,该层会自动把图像转为单通道灰度图,图像深度也就保持为1了。
修改后的完整配置代码如下:
layers { name: "data" type: IMAGE_DATA top: "data" top: "label" include { phase: TRAIN } image_data_param { source: "train.txt" batch_size: 100 new_height: 28 new_width: 28 is_color: false # 添加这个参数,强制读取单通道灰度图 } }
配置修改完成后,启动训练时可以通过日志确认输入数据的维度,正常情况下data层的输出维度会变为(100, 1, 28, 28),这就说明图像深度已经成功设置为1了。
内容的提问来源于stack exchange,提问作者user61455
相关产品推荐
相关产品推荐

