当我在自己的数据集中应用转移学习时，我自己定义input_size

Question

我正在使用tensorflow进行深度学习。

我想尝试在我自己的数据集中进行转移学习，并从tensorflow的网站下载了inceptionv3模型。我也找到了一个演示，但我发现模型input_size是299 * 299 * 3。我想自己定义input_size。因为Keras的初始v3模型可以自己定义input_size。例如input_size是512 * 512 * 3.

我不使用调整大小功能。

我尝试执行以下操作： enter image description here

但我收到以下错误： enter image description here

当我将其更改为299 * 299 * 3时，代码正常运行。

Answer 1

您无法轻易更改训练模型的输入大小。训练模型的权重仅知道如何处理此大小的输入。如果您想使用预先训练过的砝码，最佳选择是将图像尺寸调整为预期尺寸。

Answer 2

考虑到InceptionV3，您可以使用任何图像大小和张量流的预处理来获取图像大小。 Tensorflow的官方初始模块包括https://github.com/tensorflow/models/blob/master/research/inception/inception/image_processing.py，您可以在其中指定要使用的图像的大小。然后可以使用这个新尺寸重新训练模型。