python - 不同尺寸图像的预测

Question

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

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

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

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

Answer 1

神经网络是一堆神经元层，输入层的神经元数量取决于输入的形状，并且在输入层根据该形状分配权重。因此，一旦网络经过训练，您就无法更改输入大小/形状。

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