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



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

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

我不使用调整大小功能。

我尝试执行以下操作: 在此处输入图像描述

但我收到以下错误: 在此处输入图像描述

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

您无法轻松更改已训练模型的输入大小。训练模型的权重只知道如何处理这种大小的输入。如果要使用预先训练的权重,最好的选择是将图像大小调整为预期大小。

就 InceptionV3 而言,您可以使用任何图像大小,Tensorflow 的预处理将采用图像大小。Tensorflow的官方初始模块包括 https://github.com/tensorflow/models/blob/master/research/inception/inception/image_processing.py,您可以在其中指定要使用的图像的大小。然后可以使用此新大小重新训练模型。

最新更新