使用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