是否可以在keras中训练多种图像尺寸?

时间:2017-02-20 22:58:35

标签: python machine-learning tensorflow deep-learning keras

Keras将numpy数组作为训练数据的输入,但是可以创建可以采用可变输入大小的模型。我想知道是否有办法将各种尺寸的图像合并到模型的训练数据中。

1 个答案:

答案 0 :(得分:1)

您不能提供可变大小的图像来训练Keras中的模型。根据Keras API,Input图层函数如下所示。

Input(shape=(3,None,None))

其中,3表示RGB图像的通道数。但你必须清楚地告诉Keras,训练图像的宽度和高度是多少。因此,Keras无法处理可变长度的图像。

因此,您必须先将图像转换为特定尺寸,然后使用Keras训练模型。