Research Article
Research on Defective Apple Detection Based on Attention Module and ResNet-50 Network
@INPROCEEDINGS{10.1007/978-3-031-50580-5_26, author={Lei Zhao and Zhenhua Li and Qinjun Zhao and Wenkong Wang and Rongyao Jing and Kehua Du and Shijian Hu}, title={Research on Defective Apple Detection Based on Attention Module and ResNet-50 Network}, proceedings={Multimedia Technology and Enhanced Learning. 5th EAI International Conference, ICMTEL 2023, Leicester, UK, April 28-29, 2023, Proceedings, Part IV}, proceedings_a={ICMTEL PART 4}, year={2024}, month={2}, keywords={Defective Apple Detection ResNet-50 Attention Module LeakyReLU Activation Function}, doi={10.1007/978-3-031-50580-5_26} }- Lei Zhao
Zhenhua Li
Qinjun Zhao
Wenkong Wang
Rongyao Jing
Kehua Du
Shijian Hu
Year: 2024
Research on Defective Apple Detection Based on Attention Module and ResNet-50 Network
ICMTEL PART 4
Springer
DOI: 10.1007/978-3-031-50580-5_26
Abstract
In defective apple detection, stem and calyx are easily confused with defects, and the detection accuracy of defective apples is lower. In order to solve these problems, this paper proposes a defective apple detection algorithm based on attention module and ResNet-50 network. CAM attention module and LeakyReLU activation function are used to optimize ResNet-50 network, which is named as C-ResNet-50 network. During network training, we use the cosine attenuation learning rate method, which effectively reduces the oscillation of training loss and accelerates the speed of network convergence. After the training and validation of the C-ResNet-50 network, the detection accuracy of defective apples reaches 97.35%, which is 2.33% higher than that of unimproved ResNet-50 network, 3.16% higher than VGGNet network and 4.14% higher than AlexNet network. This proves that the C-ResNet-50 network can improve the accuracy of defective apple detection.
