“ValueError:输入0与图层cropping2d_1不兼容:预期ndim = 4,发现ndim = 3”

时间:2018-03-13 13:42:12

标签: python python-3.x tensorflow keras

我将单色通道图像添加到数组

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))))

0 个答案:

没有答案