https://www.ahs.ac.cn/images/0513-353X/images/top-banner1.jpg|#|苹果
https://www.ahs.ac.cn/images/0513-353X/images/top-banner2.jpg|#|甘蓝
https://www.ahs.ac.cn/images/0513-353X/images/top-banner3.jpg|#|菊花
https://www.ahs.ac.cn/images/0513-353X/images/top-banner4.jpg|#|灵芝
https://www.ahs.ac.cn/images/0513-353X/images/top-banner5.jpg|#|桃
https://www.ahs.ac.cn/images/0513-353X/images/top-banner6.jpg|#|黄瓜
https://www.ahs.ac.cn/images/0513-353X/images/top-banner7.jpg|#|蝴蝶兰
https://www.ahs.ac.cn/images/0513-353X/images/top-banner8.jpg|#|樱桃
https://www.ahs.ac.cn/images/0513-353X/images/top-banner9.jpg|#|观赏荷花
https://www.ahs.ac.cn/images/0513-353X/images/top-banner10.jpg|#|菊花
https://www.ahs.ac.cn/images/0513-353X/images/top-banner11.jpg|#|月季
https://www.ahs.ac.cn/images/0513-353X/images/top-banner12.jpg|#|菊花

Acta Horticulturae Sinica ›› 2024, Vol. 51 ›› Issue (2): 396-410.doi: 10.16420/j.issn.0513-353x.2023-0859

• New Technologies·New Methods • Previous Articles     Next Articles

Detecting Tomato Fruit Ripeness and Appearance Quality Based on Improved YOLOv5s

SUN Yuchao1,2, LI Shouhao1,2, XIA Xiubo1,3,*(), YANG Wei2,*(), LI Minzan2, ZHANG Huanchun1,3   

  1. 1 Yantai Academy of Agricultural Sciences in Shandong Province,Yantai,Shandong 265500,China
    2 Key Laboratory of Smart Agriculture Systems,Ministry of Education,China Agricultural University,Beijing 100083,China
    3 Yantai Smart Agriculture Research Center,Yantai,Shandong 265500,China
  • Received:2023-10-30 Revised:2023-12-29 Online:2024-02-25 Published:2024-02-27
  • Contact: XIA Xiubo, YANG Wei

Abstract:

The study was conducted on the detection of tomato fruit maturity and appearance quality based on deep learning methods using pink tomato as the experimental material. Two thousand and thirty-six tomato image data were collected and amplified to 5 316 through preprocessing. Then,the data was annotated and converted into files to construct an experimental dataset. The experiment improves the accuracy of SC-YOLOv5s by adding CA attention mechanism,replacing the Stem block structure,optimizing the detection layer scale based on recognition requirements,and replacing the K-means++ clustering algorithm to improve the model’s feature expression ability. By adding a fire module structure to SC-YOLOv5s for lightweight convolution and reducing the parameter count of the Bottleneck module,the SC-YOLOv5s-lite lightweight design is achieved,improving the detection speed of the model on hardware;Train and optimize the SC-YOLOv5s-lite model on the training set. The results showed that the memory usage of the SC-YOLOv5s-lite model was 7.73 M,with an accuracy rate of 89.04%,a recall rate of 83.35%,an average accuracy of 91.34%,and a detection time of 143 ms. Compared to YOLOv5s,the model parameter quantity is reduced by 54.57%,model size is compressed by 44.86%,with an average accuracy improvement of 3.98%,and the detection time is reduced by 20.99%. It has obvious advantages and is more suitable for hardware deployment.

Key words: tomato, maturity, appearance quality, detection, deep learning, computer vision