3d numpy数组到灰度图片

时间:2017-10-07 12:46:40

标签: python arrays image numpy grayscale

提取cifar10数据集时遇到的数据尺寸为32x32x3。

我可以用例如彩色图像绘制图像。 plt.imshow(train_data [2]);有什么常见的方法将数组转换为具有灰度值的维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)

与此同时,我只是保存图像并再次阅读,但我想有一种更优雅的方式将图片直接存储为灰度。

1 个答案:

答案 0 :(得分:0)

您可以使用当前的3D阵列使用matplotlib' imshow绘制灰度图像,如here所述。

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