python - 在 Python 中使用 MNIST FASHION 为图像分类准备/渲染图像

Question

我知道以前有人问过这个问题，但我仍然面临一些问题。

建立神经网络并训练模型后，我现在想对桌面上的图像进行分类。出于这个原因，图像必须在监督学习之前准备好……</p>

如何将普通图片转换为 (1, 28, 28) 格式？

我试着这样做

Img = imageio.imread(f‘path/pic.png‘)
Image = numpy.expand(Img, 0)
Print(Image.shape) RETURNS (1, 28, 28, 3) and NOT (1, 28, 28)

任何想法，灵感，......在此先感谢

标签： pythonpython-3.ximagerenderingimage-classification

Answer 1

您可以不使用imageio库，而是使用 OpenCV，即cv2库（需要先安装）。

import numpy as np
import cv2

Img = cv2.imread('path/pic.png', 0)  # Need to pass in the zero as a flag to be read in gray-scale
Image = np.expand_dims(Img, 0)
print(Image.shape)

> (1, m, n)