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 Technology and 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,and ZHANG Huanchun1,3   

  1. 1Yantai Academy of Agricultural Sciences in Shandong Province,Yantai,Shandong 265500,China;2Key Laboratory of Smart Agriculture Systems,Ministry of Education,China Agricultural University,Beijing 100083,China;3Yantai Smart Agriculture Research Center,Yantai,Shandong 265500,China
  • Online:2024-02-25 Published:2024-02-27

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

CLC Number: