Welcome to Francis Academic Press

Academic Journal of Computing & Information Science, 2024, 7(8); doi: 10.25236/AJCIS.2024.070802.

Research on image classification based on ResNet

Author(s)

Huimin Lu1, Handong Song2

Corresponding Author:
Huimin Lu
Affiliation(s)

1School of Artificial Intelligence and Automation, Huazhong University of Science and Technology, Wuhan, 430074, China

2School of Mechanical Engineering, University of Jinan, Jinan, 250024, China

Abstract

As one of the representative algorithms of convolutional neural networks, deep residual network has achieved very good results in computer vision, classification and other fields. Deep residual networks redefine the way networks learn, allowing networks to directly learn the difference between input and output information. This paper takes the deep residual network as the main research object, introduces the innovation of the deep residual network, and uses the deep residual network and other neural network models to train on different datasets. It is found that the highest accuracy of the test set of the deep residual network on the MNIST dataset is 0.555% higher than that of LeNet. 0.225% higher than AlexNet; The maximum accuracy of the test set on the CIFAR-10 dataset is 23.13% higher than LeNet and 2.46% higher than AlexNet.

Keywords

Deep residual network, Convolutional neural network, Image recognition, Neural network model

Cite This Paper

Huimin Lu, Handong Song. Research on image classification based on ResNet. Academic Journal of Computing & Information Science (2024), Vol. 7, Issue 8: 9-14. https://doi.org/10.25236/AJCIS.2024.070802.

References

[1] Li X , Zhang J , Xue Y ,et al.Classification of hops image based on ResNet-ConvLSTM and research of intelligent liquor picking system[J].Measurement, 2022(194-):194. DOI:10.1016/j. measurement. 2022. 110955.

[2] Xu Z , Sun K , Mao J .Research on ResNet101 Network Chemical Reagent Label Image Classification Based on Transfer Learning[C]//2020 IEEE 2nd International Conference on Civil Aviation Safety and Information Technology (ICCASIT).IEEE, 2020.DOI:10.1109/ICCASIT50869.2020.9368658.

[3] Xu Jiaqin, Zhang Min, Wang Liuying, et al. Application of deep learning method based on residual neural network in ECG automatic recognition of arrhythmia [J]. Chinese Journal of Health Statistics, 2023, 40(3):345-348.

[4] Han Zhengjie, Niu Rongjun, Ma Zikui, et al. Bearing Fault Diagnosis Method based on improved residual neural network based on attention Mechanism [J]. Journal of Vibration and Shock, 2023, 42(16):82-91.

[5] Chen Chao, Qi Feng. Review on the development of Convolutional neural networks and their applications in computer vision [J]. Computer science, 2019, 46 (3): 11. DOI: 10.11896 / j.i SSN. 1002-137 - x. 2019.03.008. 

[6] Zeng F , Hu W , He G ,et al.Imbalanced Thangka Image Classification research Based on the ResNet Network [J].Journal of Physics: Conference Series, 2021, 1748(4):042054 (8pp).DOI:10.1088/1742-6596/1748/4/042054.

[7] Deng Changyin, Zhang Jie. Handwritten digit Recognition based on improved LeNet-5 model [J]. Information and communications, 2018 (1): 4. DOI: 10.3969 / j.i SSN. 1673-1131.2018.01.047.

[8] Wang Z .Research on Garbage Image Classification Based on Neural Network[J].2022 International Conference on Computer Network, Electronic and Automation (ICCNEA), 2022:214-217.DOI: 10. 1109/ ICCNEA57056.2022.00055.