对不同尺寸图像的预测

Question

是否可以 运行 对与训练数据大小不同的图像进行预测？我在一定尺寸（1000*1000）的图片上训练了一个unet_learner，导出来备用。

然后我用以下内容加载它：

learn = load_learner(modelpath)
preds = learn.predict(image) 

但是如果输入图像的尺寸不同会怎样？

Answer 1

神经网络是一堆神经元层，输入层的神经元数量取决于输入的形状，权重根据输入层的形状分配。所以一旦网络被训练，你就不能改变输入size/shape。

因此，在将输入馈送到网络之前，调整图像大小并将其转换为 numpy 数组，根据所需的输入尺寸重新整形数组。