Should work for Keras backend set to Theano

[maximus009]

Currently, the backend and image dimension ordering is set to work only with Tensorflow (tf) and will not work if set to Theano (th). Need to remove that with a condition check and and automatic reordering of the tensor.