使用scipy.misc.imread()函數(shù)對齊預(yù)測輸出和ground truth的通道。

替換新的數(shù)據(jù)輸入模型，經(jīng)常會(huì)報(bào)數(shù)據(jù)維度不對的問題。這是由于預(yù)測數(shù)據(jù)和ground truth數(shù)據(jù)通道數(shù)沒有對齊。

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

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

‘L’（8位像素，黑白）#單通道
‘P’（8位像素，使用調(diào)色板映射到任何其他模式）#單通道
‘RGB’（3x8位像素，真彩色）#三通道
‘RGBA’（4x8位像素，帶透明蒙版的真彩色）
‘CMYK’（4x8位像素，分色）
‘YCbCr’（3x8位像素，彩色視頻格式）
‘I’（32位有符號整數(shù)像素）
‘F’（32位浮點(diǎn)像素）

單通道轉(zhuǎn)換三通道

image = np.expand_dims(a, axis=2)
image = np.concatenate((image, image, image), axis=-1) #等價(jià)于axis=2