Journal of Optoelectronics · Laser, Volume. 33, Issue 6, 643(2022)

ResNet fruit appearance quality classification based on dual channel attention mechanism

ZHAO Hui1,2, QIAO Yanjun1、*, WANG Hongjun1, and YUE Youjun1
Author Affiliations
  • 1[in Chinese]
  • 2[in Chinese]
  • show less

    In order to realize the fast and accurate appearance quality classification of picked fruits,and cooperate with the sorting production line to complete the large-scale centralized sorting of fruits,a fruit classification method based on improved ResNet is proposed in this study.Firstly,the residual module in ResNet network is combined with the dual channel squeeze-and-excitation block (DC-SE Block) to enhance the effective channel features,suppress the inefficient or invalid channel features,and improve the expression ability of the feature map,so as to improve the recognition accuracy.Secondly,the Inception module is added to the original ResNet model to fuse the characteristics of different scales of fruit,so as to enhance the recognition ability of small defects.Finally,four kinds of fruit images with different appearance quality are enhanced,and the model is initialized by transfer learning method.Taking apple as an example,the experimental results show that the accuracy of the improved model trained by the data set is 99.7%,which is higher than 98.5% of the original model;The precision rate is 99.7%,which is higher than 98.3% of the original model;The recall rate reaches 99.7%,which is higher than 98.7% of the original model; The average detection speed under graphic processing unit (GPU) is 32.3 frame/s,which is slightly lower than 35.7 frame/s of the original model.Compared with several advanced classification methods such as GoogleNet and MobileNet,and compared with different improved models,the results show that the proposed method has good classification performance,and has important reference value for solving the problem of accurate classification of fruit appearance quality.

    Tools

    Get Citation

    Copy Citation Text

    ZHAO Hui, QIAO Yanjun, WANG Hongjun, YUE Youjun. ResNet fruit appearance quality classification based on dual channel attention mechanism[J]. Journal of Optoelectronics · Laser, 2022, 33(6): 643

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Received: Sep. 6, 2021

    Accepted: --

    Published Online: Oct. 9, 2024

    The Author Email: QIAO Yanjun (qiaoyanjun1996@163.com)

    DOI:10.16136/j.joel.2022.06.0633

    Topics