Arsey/keras-transfer-learning-for-oxford102

Is some change needed in resnet50.py? Error while creating model

shirishr opened this issue · 1 comments

I get error:
Negative dimension size caused by subtracting 3 from 2 for 'max_pooling2d_4/MaxPool' (op: 'MaxPool') with input shapes: [?,2,112,64].
I am using Python 3.5 and Tensorflow 1.2 keras 2.0.2

I tried various combinations of values for:
image_dim_ordering and
image_data_format the errors would be different but still related to dim mismatch.
So I abandoned Tensorflow and switched to Theano backend with

keras.backend.set_image_dim_ordering('tf')
keras.backend.set_image_data_format('channels_first')

This combination seems to work. The requirements.txt file in the Github specifically calls out theano (0.9.0) Hence closing the issue