使用scipy.misc.imread()函数对齐预测输出和ground truth的通道。

替换新的数据输入模型,经常会报数据维度不对的问题。这是由于预测数据和ground truth数据通道数没有对齐。

scipy.misc.imread(name, flatten=False, mode=None )

mode可以是以下字符串之一:

‘L’(8位像素,黑白)#单通道
‘P’(8位像素,使用调色板映射到任何其他模式)#单通道
‘RGB’(3x8位像素,真彩色)#三通道
‘RGBA’(4x8位像素,带透明蒙版的真彩色)
‘CMYK’(4x8位像素,分色)
‘YCbCr’(3x8位像素,彩色视频格式)
‘I’(32位有符号整数像素)
‘F’(32位浮点像素)

 

单通道转换三通道

image = np.expand_dims(a, axis=2)
image = np.concatenate((image, image, image), axis=-1) #等价于axis=2
posted on 2021-05-20 00:18  Google-boy  阅读(268)  评论(0编辑  收藏  举报