对不同尺寸的图像进行预测

时间:2019-08-16 09:29:33

标签: python machine-learning deep-learning fast-ai

是否可以对尺寸与训练数据不同的图像进行预测?我已经在一定尺寸(1000 * 1000)的图片上训练了unet_learner,并将其导出以供以后使用。

然后我将其加载以下内容:

item

但是,如果输入图像的大小不同会怎样?

1 个答案:

答案 0 :(得分:1)

神经网络是神经元层的堆栈,输入层中神经元的数量取决于输入的形状,权重是根据输入层的形状分配的。因此,一旦训练好网络,您将无法更改输入大小/形状。

因此,在将输入提供给网络之前,请调整图像大小并将其转换为numpy数组,然后根据所需的输入尺寸对数组进行整形。