Able to run the learning for translation, but rotation is still an issue.

Able to run the learning for translation, but rotation is still an
issue. Need to debug how can we fix rotation of image learning part.