Dragon fruit weight estimation based on machine vision and machine learning
CSTR:
Author:
Affiliation:

(1. College of Engineering, South China Agricultural University, Guangzhou, Guangdong 510642, China; 2. Guangdong Provincial Key Laboratory of Agricultural Artificial Intelligence, Guangzhou, Guangdong 510642, China; 3. Key Laboratory of Key Technology on Agricultural Machine and Equipment, Ministry of Education, Guangzhou, Guangdong 510642, China)

Clc Number:

Fund Project:

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

    Objective: In order to solve the problem of manual weighting of dragon fruit, including time-consuming, laborious and expensive, an automated weight estimation method based on machine vision and machine learning was proposed in this research. Methods: Firstly, 106 dragon fruits were weighed, recorded and photographed, and images of dragon fruits were constructed. Secondly, binary images were obtained after denoising and segmentation. Moreover, the three features of pixel area, major axis pixel length and minor axis pixel length of dragon fruits were extracted on the basis of binary images. The three features of each image and their corresponding weights were combined into a set of data, which was divided into training set and test set according to the ratio of 7∶3. Finally, the training set was input into the Gradient Boosting, Random Forest, K-Neighbors and Artificial Neural Networks machine-learning models for training, and the test sets were used for model evaluation. Results: The evaluation index of the Artificial Neural network performed well compared with other models, with R2 of 0.986 and RMSE of 13.091. Conclusion: The experimental result demonstrates that the method proposed in this research can accomplish the weight estimation of dragon fruit effectively, and meet the weight estimation requirements of dragon fruit.

    Reference
    Related
    Cited by
Get Citation

梁英凯,商枫楠,陈 桥,等.基于机器视觉与机器学习的火龙果重量估计[J].食品与机械英文版,2023,39(7):99-103.

Copy
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:November 17,2022
  • Revised:
  • Adopted:
  • Online: October 20,2023
  • Published:
Article QR Code