Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

What size should my image be to retrain Inception?

I am following codelabs Tensorflow for poets guide to re-train inception v3 with my own images. But there is no mention of what the size my image should be. I also watched some Youtube video that suggested cropping and filling in white spaces to make square images. But it didn't really mention the size.

How should I resize my training images to so I get the best result re-training inception?

like image 601
Learning C Avatar asked Dec 23 '22 21:12

Learning C


1 Answers

The code does the resizing for you. Have a look at retrain.py. I have listed the code responsible for deciding the size of the images depending on the network architecture.

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'
like image 109
Kaushalya Avatar answered Dec 30 '22 22:12

Kaushalya