我将单色通道图像添加到数组
image = cv2.cvtColor(cv2.imread(path), cv2.COLOR_BGR2HSV)
tmp = image[:,:,1].reshape(160, 320, 1)
images.append(tmp)
但是,在顺序模型开始时,我正在
ValueError:输入0与layer cropping2d_1:expected不兼容 ndim = 4,发现ndim = 3
Cropping2D
如何适用于单色通道图像?
模型开始如下:
model = Sequential()
model.add(Lambda(lambda x: x/255.0 - 0.5, input_shape=(160, 320, 1)))
model.add(Cropping2D(cropping=((50, 20), (0, 0))))