vgg16.npy 进行fine-tune的时候，输入图像大小不是224，全连接层维度不对，怎么解决呢？

Question

问题描述

vgg16.npy 进行fine-tuning的时候，由于输入图像的维度不是224的，导致在展平之后，全连接层的维度不对，请问这个问题怎么解决呢？

问题出现的环境背景及自己尝试过哪些方法

我想用imagenet已经训练好的模型进行fine-tune，训练自己的模型，听别人说可以 resize 成 224×224 大小的图像，但是不知道大佬们是如何过的呢？

Answer 1

最直接的方法是

image = transform.resize(image, (w,h))