我正在遵循 代码实验室Tensorflow的诗人指南,用自己的图像重新训练Inception v3。但是没有提及我的图像应该是多大。我还观看了一些Youtube视频,该视频建议裁剪并填充白色空间以制作方形图像。但这并没有真正提到尺寸。
我应该如何调整训练图像的大小以使重新训练开始时获得最佳结果?
该代码为您调整大小。看看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'