Abstract
A machine needs to recognize orientation in an image to address various rotation related problems. To calculate this rotation, one must require the information about different objects that present into the image. Hence this becomes a pattern recognition task. By using Deep Learning this issue of calculation of image rotation can be addressed as deep learning possess excellent ability of feature extraction. This paper proposes a novel deep learning-based approach to estimate the angle of rotation very efficiently. Kaggle dataset (Rotated Coins) and Caltech-256 has been used for this research, but the data available was limited hence this research utilize data augmentation by rotating the given dataset at random angles. Initially the unlabeled image has been rotated at different angles and store the values to be used as training dataset. Finally at the output a regression layer has been used to identify the angle of rotation for input image. The proposed deep learning approach provides a better result in terms of validation parameters like R-square, MSE, MAE. With proposed approach the value of R-square, MSE, and MAE for Kaggle dataset (Rotated Coins) obtained is 0.9846, 0.0013 and 0.0127 respectively. While for Caltech-256 Dataset proposed approach reported R-square, MSE, and MAE of 0.9503, 0.0039 and 0.0240 respectively. The proposed approach also helps in finding the position of an object by calculating the angle of rotation in an image.
This is a preview of subscription content, log in via an institution to check access.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Vailaya, A., Zhang, H., Yang, C., Liu, F.I., Jain, A.K.: Automatic image orientation detection. IEEE Trans. Image Process. 11, 746–755 (2002). https://doi.org/10.1109/TIP.2002.801590
Ciocca, G., Cusano, C., Schettini, R.: Image orientation detection using LBP-based features and logistic regression. Multimed. Tools App. 74(9), 3013–3034 (2013). https://doi.org/10.1007/s11042-013-1766-4
Fefilatyev, S., Smarodzinava, V., Hall, L.O., Goldgof, D.B.: Horizon detection using machine learning techniques. In: Proceedings - 5th International Conference on Machine Learning and Applications, ICMLA 2006, pp 17–21 (2006)
Workman, S., Zhai, M., Jacobs, N.: Horizon lines in the wild. In: British Machine Vision Conference 2016, BMVC 2016. British Machine Vision Conference, BMVC, pp. 20.1–20.12 (2016)
Ávila, B.T., Lins, R.D.: A fast orientation and skew detection algorithm for monochromatic document images (2005)
Amin, A., Fischer, S.: A document skew detection method using the Hough transform (2000)
Huang, K., Chen, Z., Yu, M., Yan, X., Yin, A.: An efficient document skew detection method using probability model and Q test. Electronics (Switzerland) 9, 55 (2020). https://doi.org/10.3390/electronics9010055
Ogiue, S., Ito, H.: Method of correction of rotated images using deep learning networks. In: Proceedings - 2018 7th International Congress on Advanced Applied Informatics, IIAI-AAI 2018. Institute of Electrical and Electronics Engineers Inc., pp. 980–981 (2018)
Gidaris, S., Singh, P., Komodakis, N.: Unsupervised representation learning by predicting image rotations (2018)
Zhou, Y., Shi, J., Yang, X., Wang, C., Wei, S., Zhang, X.: Rotational objects recognition and angle estimation via kernel-mapping cnn. IEEE Access 7, 116505–116518 (2019). https://doi.org/10.1109/ACCESS.2019.2933673
Mironica, I., Zugravu, A.: A fast deep learning network for automatic image auto-straightening (2021)
Rotated Coins | Kaggle. https://www.kaggle.com/competitions/coins. Accessed 1 Aug 2022
Caltech 256 Image Dataset | Kaggle. https://www.kaggle.com/datasets/jessicali9530/caltech256. Accessed 1 Aug 2022
Lee, C.-Y., Gallagher, P.W., Tu, Z.: Generalizing pooling functions in convolutional neural networks: mixed, gated, and tree (2015)
Worrall, D.E., Garbin, S.J., Turmukhambetov, D., Brostow, G.J.: harmonic networks: deep translation and rotation equivariance (2017)
Srivastava, N., Hinton, G., Krizhevsky, A., Salakhutdinov, R.: Dropout: a simple way to prevent neural networks from overfitting (2014)
Kingma, D.P., Ba, J.: Adam: a method for stochastic optimization (2014)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2023 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Rane, T., Bhatt, A. (2023). A Deep Learning-Based Regression Scheme for Angle Estimation in Image Dataset. In: Santosh, K., Goyal, A., Aouada, D., Makkar, A., Chiang, YY., Singh, S.K. (eds) Recent Trends in Image Processing and Pattern Recognition. RTIP2R 2022. Communications in Computer and Information Science, vol 1704. Springer, Cham. https://doi.org/10.1007/978-3-031-23599-3_21
Download citation
DOI: https://doi.org/10.1007/978-3-031-23599-3_21
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-23598-6
Online ISBN: 978-3-031-23599-3
eBook Packages: Computer ScienceComputer Science (R0)