1

我正在尝试将维度 (32, 32, 3, 10000) 维度数据集的数据集转换为灰度数据集,其中我将具有 (32, 32, 1, 10000) 维度,但我需要有 1 个通道,因为我会将其输入到神经网络。我尝试使用 numpy.average,但形状变为 (32, 32, 10000),TensorFlow 单元没有将其作为输入。我什至尝试手动对其进行平均,但结果相同。你们能帮我解决这个问题吗?

4

2 回答 2

0

(32, 32, 1000)可以在获得形状后添加您需要的额外尺寸。

您可以尝试np.expand_dims使用axis参数来定义您希望额外的“1”出现的位置。

于 2021-08-19T10:59:28.940 回答
0

尝试 (0.299 * Red) + (0.587 * Green) + (0.114 ∙ Blue)而不是平均。

于 2021-08-19T10:55:05.763 回答