Method for Fruit and Vegetable Automatic Recognition Based on Residual Block and Attention Mechanism
CSTR:
Author:
Affiliation:

Clc Number:

Fund Project:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    To solve the problems of low efficiency and high cost in fruits and vegetables recognition, a fruit and vegetable recognition model based on residual block and attention mechanism was proposed, and successfully deployed in fruit and vegetable intelligent recognition equipment. The fruit and vegetable automatic recognition device was composed of Raspberry Pi, STM32F103ZET6, camera, weighing sensor, processor, display screen, micro printer, binding machine and power supply. The central controller interacted with the display screen to display various parameters in real time. The image and quality of the object to be measured were collected through the camera and weighing sensor. The fruit and vegetable automatic recognition model deployed in the Raspberry Pi could accurately identify the fruits and vegetables. At the same time, it cooperated with MCU STM32F103ZET6 to print fruit and vegetable related information and control the tying machine to seal and pack. Based on YOLO v5 network, an automatic recognition model RB+CBAM-YOLO v5 was constructed by adding residual blocks and attention mechanism. The network was trained with the self-made data set, and six kinds of networks were compared, and the optimal network was selected for the device side detection test. The experimental results showed that the accuracy rate, recall rate and mAP0.5 of RB+CBAM-YOLO v5 were 83.55%, 96.08% and 96.20%, respectively, which were 4.47 percentage points, 1.10 percentage points and 0.90 percentage points higher than those of YOLO v5. The RB+CBAM-YOLO v5 model was deployed in the embedded device Raspberry Pi, and the device could realize accurate identification, automatic weighing, printing slip and fast packaging functions, which could meet the needs of fruits and vegetables identification and unsold devices.

    Reference
    Related
    Cited by
Get Citation
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:June 30,2023
  • Revised:
  • Adopted:
  • Online: August 30,2023
  • Published:
Article QR Code