python - 如何在卷积神经网络和自动编码器中仅对灰度图像使用两个通道而不是 3 通道 RGB

Question

我有一个灰度图像（超声波），我想使用 CNN 和自动编码器来训练它们，但是我有默认通道数为 3 的问题，当我使用 3 个通道时报告错误，我怎么能只使用 2 个通道.

ValueError: Dimensions must be equal, but are 2 and 3 for 'loss_mse/sub' (op: 'Sub') with input shapes: [?,128,128,2], [?,128,128,3].

标签： pythondeep-learningautoencoder