Transfer Learning for Automatic Image Orientation Detection Using Deep Learning and Logistic Regression
The number of images produced each day increased significantly. The ability to detect and correct an image’s orientation can provide several advantages in computer vision. This paper presents a new framework based on a transfer learning technique for automatically detecting image orientat...
| Main Authors: | , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
IEEE
2022-01-01
|
| Series: | IEEE Access |
| Subjects: | |
| Online Access: | https://ieeexplore.ieee.org/document/9965403/ |
| _version_ | 1811177587226968064 |
|---|---|
| author | Ayoub Benali Amjoud Mustapha Amrouch |
| author_facet | Ayoub Benali Amjoud Mustapha Amrouch |
| author_sort | Ayoub Benali Amjoud |
| collection | DOAJ |
| description | The number of images produced each day increased significantly. The ability to detect and correct an image’s orientation can provide several advantages in computer vision. This paper presents a new framework based on a transfer learning technique for automatically detecting image orientation. To implement the power of deep neural networks, we applied a convolutional neural network model pre-trained on the ImageNet database for feature extraction. Then, we built a multi-class logistic regression classifier to detect the four image orientation probabilities corresponding to the following orientations (0 for no orientation, 90, 180, and 270). We tested our model on the SUN-397 dataset, one of the most extensive data sets currently used for image-orientation detection tasks. We conducted a cross-dataset evaluation for in-depth testing and analysis. We also examined our model using different old and recent state-of-the-art convolutional neural network (CNN) baselines. We demonstrate that our model yields promising results based on transfer learning for feature extraction combined with a one-vs-rest logistic regression classifier. Our proposed model surpassed the state-of-the-art results in terms of accuracy and performance. |
| first_indexed | 2024-04-11T06:04:56Z |
| format | Article |
| id | doaj.art-23f4478ee5784bcd939a6855da3ae2d0 |
| institution | Directory Open Access Journal |
| issn | 2169-3536 |
| language | English |
| last_indexed | 2024-04-11T06:04:56Z |
| publishDate | 2022-01-01 |
| publisher | IEEE |
| record_format | Article |
| series | IEEE Access |
| spelling | doaj.art-23f4478ee5784bcd939a6855da3ae2d02022-12-22T04:41:32ZengIEEEIEEE Access2169-35362022-01-011012854312855310.1109/ACCESS.2022.32254559965403Transfer Learning for Automatic Image Orientation Detection Using Deep Learning and Logistic RegressionAyoub Benali Amjoud0https://orcid.org/0000-0002-0903-2602Mustapha Amrouch1IRF-SIC Laboratory, Faculty of Sciences, Ibn Zohr University, Agadir, MoroccoIRF-SIC Laboratory, Faculty of Sciences, Ibn Zohr University, Agadir, MoroccoThe number of images produced each day increased significantly. The ability to detect and correct an image’s orientation can provide several advantages in computer vision. This paper presents a new framework based on a transfer learning technique for automatically detecting image orientation. To implement the power of deep neural networks, we applied a convolutional neural network model pre-trained on the ImageNet database for feature extraction. Then, we built a multi-class logistic regression classifier to detect the four image orientation probabilities corresponding to the following orientations (0 for no orientation, 90, 180, and 270). We tested our model on the SUN-397 dataset, one of the most extensive data sets currently used for image-orientation detection tasks. We conducted a cross-dataset evaluation for in-depth testing and analysis. We also examined our model using different old and recent state-of-the-art convolutional neural network (CNN) baselines. We demonstrate that our model yields promising results based on transfer learning for feature extraction combined with a one-vs-rest logistic regression classifier. Our proposed model surpassed the state-of-the-art results in terms of accuracy and performance.https://ieeexplore.ieee.org/document/9965403/Transfer learningdeep learningconvolutional neural networksautomatic detectionimage orientationlogistic regression |
| spellingShingle | Ayoub Benali Amjoud Mustapha Amrouch Transfer Learning for Automatic Image Orientation Detection Using Deep Learning and Logistic Regression IEEE Access Transfer learning deep learning convolutional neural networks automatic detection image orientation logistic regression |
| title | Transfer Learning for Automatic Image Orientation Detection Using Deep Learning and Logistic Regression |
| title_full | Transfer Learning for Automatic Image Orientation Detection Using Deep Learning and Logistic Regression |
| title_fullStr | Transfer Learning for Automatic Image Orientation Detection Using Deep Learning and Logistic Regression |
| title_full_unstemmed | Transfer Learning for Automatic Image Orientation Detection Using Deep Learning and Logistic Regression |
| title_short | Transfer Learning for Automatic Image Orientation Detection Using Deep Learning and Logistic Regression |
| title_sort | transfer learning for automatic image orientation detection using deep learning and logistic regression |
| topic | Transfer learning deep learning convolutional neural networks automatic detection image orientation logistic regression |
| url | https://ieeexplore.ieee.org/document/9965403/ |
| work_keys_str_mv | AT ayoubbenaliamjoud transferlearningforautomaticimageorientationdetectionusingdeeplearningandlogisticregression AT mustaphaamrouch transferlearningforautomaticimageorientationdetectionusingdeeplearningandlogisticregression |