Question

我输入的图像尺寸为（28，28，3）。我用几幅尺寸为（28、28、1）的图像训练了一个keras模型。我想要使用此模型检查单个测试图像，但是每次出现尺寸错误时，如何将原始尺寸（28，28，3）减小为（28，28，1）？

test_image = image.load_img('test/number3.png' , target_size = (28, 28))
test_image = image.img_to_array(test_image)
test_image = np.expand_dims(test_image, axis = 1)
result = classifier.predict(test_image)

Answer 1

根据您要减少尺寸的方式，您可以选择这样的颜色通道之一

one_channel_image = test_image[:,:,0]

或者您可以找到各颜色通道的均值

one_channel_image = np.mean(test_image, axis=2)

根据我对ML图像问题的经验，只使用一个通道就可以了。

如果需要将尺寸从（28，28）增加到（28，28，1），可以使用numpy.reshape

one_channel_image = test_image.reshape((28, 28, 1))

减少python中的图像尺寸

1 个答案: