3d numpy数组转灰度图片

Question

在提取cifar10数据集im时，遇到了维数为32×32×3的数组。

例如，我可以用plt.imshow(Train_data2)绘制彩色图像；将数组转换为具有灰度值的32x32维度的常用方法是什么？

train_data, train_labels, test_data, test_labels = 
load_cifar10_data(data_dir)

print(train_data.shape)
print(train_labels.shape)

输出：

(50000，32，32，3) (50000，)

同时，我只是在保存图片并再次阅读，但我猜有一种更优雅的方法可以直接将图片存储为灰度。

Answer 1

您可以使用matplotlib的imshow来使用当前的3D数组来绘制灰度图像，如here所述。

import matplotlib.pyplot as plt
plt.imshow(train_data , cmap = "gray")