如何将灰度图像数据集加载到移动网络模块

Question

我试图将灰度图像数据集（时尚mnist）加载到MobileNet模型中，以预测手写数字，但根据this教程，只有RGB图像可以加载到模型中。当我试图提供时尚杂志的样本时，它给出了以下错误

Error when checking input: expected keras_layer_13_input to have shape (224, 224, 3) but got array with shape (224, 224, 1)

如何解决这个问题？在

Answer 1

Mobilenet v2需要RGB。您还可以使用PIL中的convert函数。在

试试这个：

from PIL import Image
x= Image.open(input_image).resize((96,96)).convert("RGB")

文档在这里：https://pillow.readthedocs.io/en/stable/reference/Image.html