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

时间:2018-08-31 21:22:50

标签: python deep-learning autoencoder

我有一个灰度图像(超声波),我想使用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].

0 个答案:

没有答案