对于RGB的图片读入时储存形状为(H,W,3),但是灰度图读入时的储存形状为(H,W),如果想将维度变成(H,W,1)我们可以使用如下的numpy命令来实现:
import numpy as np
train_vessels=np.expand_dims(train_vessels, axis=3)
train_vessels是读取的一张灰度图片
对于RGB的图片读入时储存形状为(H,W,3),但是灰度图读入时的储存形状为(H,W),如果想将维度变成(H,W,1)我们可以使用如下的numpy命令来实现:
import numpy as np
train_vessels=np.expand_dims(train_vessels, axis=3)
train_vessels是读取的一张灰度图片