search

d4nst / RotNet

https://d4nst.github.io/2017/01/12/image-orientation/
MIT License
539 stars 189 forks source link

correct_rotation.py problems #14

Closed swiz23 closed 6 years ago

swiz23 commented 6 years ago

I noticed that in your correct_rotation.py, - in line 16, you wrote '.jpg' twice. Is one of those supposed to be .JPG? I'm testing on some of my own pictures that end with .JPG and if I don't edit the correct_rotation.py code, I get an error message saying that 'numpy.core._internal.AxisError: axis 1 is out of bounds for array of dimension 1'. However, once I edit the code to include '.JPG', it runs with no errors.

Also, I got an error - TypeError: softmax() got an unexpected keyword argument 'axis' - if I didn't first delete 'axis=axis' in...

File "/home//tensorflow/local/lib/python2.7/site-packages/keras/backend/tensorflow_backend.py", line 3149, in softmax

I was doing some research for the 'softmax' error and I found that this only occurs in some combinations of Tensorflow and Keras versions. The ones I am using are Tensorflow 1.4.0 and Keras 2.2.0

d4nst commented 6 years ago

Thanks for reporting this. It was meant to be '.jpeg'. I have pushed a commit (67b44e0) that fixes that and the issue with upper-case extensions.

Regarding your second error, I believe you need TensorFlow version 1.5 or above if you are using Keras 2.2.0 (the axis parameter in the softmax function was introduced in TensorFlow 1.5)