我的图像应该多大才能重新训练Inception？

Question

我正在遵循 代码实验室Tensorflow的诗人指南，用自己的图像重新训练Inception v3。但是没有提及我的图像应该是多大。我还观看了一些Youtube视频，该视频建议裁剪并填充白色空间以制作方形图像。但这并没有真正提到尺寸。

我应该如何调整训练图像的大小以使重新训练开始时获得最佳结果？

Answer 1

该代码会自动为您准备图像并将其输入网络。您需要做的就是正确设置文件夹并提供足够的训练图像。根据我的经验，图像的大小并不重要。我按照说明使用 640x480 和 1280x1024 图像进行了重新训练。我通过每类 400 - 1000 张图像的训练规模获得了很好的结果。

Answer 2

该代码为您调整大小。看看retrain.py。我列出了负责根据网络体系结构确定图像大小的代码。

if architecture == 'inception_v3':
    # pylint: disable=line-too-long
    data_url = 'http://download.tensorflow.org/models/image/imagenet/inception-2015-12-05.tgz'
    # pylint: enable=line-too-long
    bottleneck_tensor_name = 'pool_3/_reshape:0'
    bottleneck_tensor_size = 2048
    input_width = 299
    input_height = 299
    input_depth = 3
    resized_input_tensor_name = 'Mul:0'