python - 图像数据 d_loss, g_loss = model_loss(input_images, input_z, data_shape[3]) 的尺寸无效

Question

当我编写下面的代码时，它给了我图像数据尺寸无效的错误。

for location in tqdm(df.iloc[:]['ID']):
    img = cv2.imread(location,0) 
    img = cv2.resize(img, (w,h), interpolation = cv2.INTER_AREA)
    img = img.reshape(w,h,1)
    left.append(img)
X1 = np.array(left)

在下面的代码中，如果我在这个 data_shape[3] 中写了 3，它会为图像数据提供错误的无效尺寸

d_loss, g_loss = model_loss(input_images, input_z, data_shape[3])

但如果我在 data_shape[2] 中写 2，则表示图像尺寸不能为 1，它必须是 2 维或 3 维 (64,64,1)

d_loss, g_loss = model_loss(input_images, input_z, data_shape[2]

请解决此问题。