Laser & Optoelectronics Progress, Volume. 61, Issue 10, 1012003(2024)

Multispectral Apple Surface Defect Detection Based on Improved YOLOv7-tiny

Chunjian Hua1,2、*, Mingchun Sun1,2, Yi Jiang1,2, Jianfeng Yu1,2, and Ying Chen3
Author Affiliations
  • 1School of Mechanical Engineering, Jiangnan University, Wuxi 214122, Jiangsu , China
  • 2Jiangsu Key Laboratory of Advanced Food Manufacturing Equipment & Technology, Wuxi 214122, Jiangsu , China
  • 3School of Internet of Things Engineering, Jiangnan University, Wuxi 214122, Jiangsu , China
  • show less

    A defect detection model based on improved YOLOv7-tiny is proposed herein to address the problem of different detection methods for different defects on apple surface. Combined with RGB+NIR multispectral images collected by a camera, various defects on the apple surface are detected and classified. First, to extract more effective feature information and improve the ability to locate defects, coordinate attention (CA) is used to aggregate coordinate information in the backbone network, and a contextual transformer (CoT) module is added behind the backbone network to increase the global receptive field. Second, it is combined with the weighted bidirectional feature pyramid to adjust the proportion of each branch in the structure to enhance the feature fusion ability of efficient layer aggregation networks. Finally, the loss function is replaced by Focal-EIoU loss to solve the problem of unbalanced samples. The mean average precision (mAP) @0.5 of the improved network increases by 1.2 percentage points to 93.2%, and the recognition speed is 89.3 frames/s. The research content of this paper provides a more efficient method for apple surface defect detection and a more accurate basis for apple grading.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Chunjian Hua, Mingchun Sun, Yi Jiang, Jianfeng Yu, Ying Chen. Multispectral Apple Surface Defect Detection Based on Improved YOLOv7-tiny[J]. Laser & Optoelectronics Progress, 2024, 61(10): 1012003

    Download Citation

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

    Category: Instrumentation, Measurement and Metrology

    Received: Aug. 10, 2023

    Accepted: Oct. 9, 2023

    Published Online: Apr. 29, 2024

    The Author Email: Chunjian Hua (277795559@qq.com)

    DOI:10.3788/LOP231895

    CSTR:32186.14.LOP231895

    Topics