我的形象应该在多大程度上重新训练开始？

Question

我正在关注 codelabs Tensorflow for poets指南用我自己的图像重新训练初始v3。但是没有提到我的形象应该是多大的。我还观看了一些Youtube视频，建议裁剪和填充白色空间以制作方形图像。但它并没有真正提到它的大小。

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

Answer 1

代码会为您调整大小。看看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'

Answer 2

代码为您自动准备图像并将其提供给网络。您需要做的就是正确设置文件夹并提供足够的训练图像。根据我的经验，图像的大小并不重要。我按照说明使用640x480和1280x1024图像进行了再培训。我的成绩很好，每班400到1000张图像。