文献类型：期刊文献

英文题名：Lightweight SM-YOLOv5 Tomato Fruit Detection Algorithm for Plant Factory

作者：Wang, Xinfa[1,2];Wu, Zhenwei[1,3,4];Jia, Meng[3];Xu, Tao[1];Pan, Canlin[1];Qi, Xuebin[4];Zhao, Mingfu[1]

第一作者：Wang, Xinfa

通讯作者：Zhao, MF[1]

机构：[1]Henan Inst Sci & Technol, Sch Informat Engn, Xinxiang 453003, Peoples R China;[2]Sumy Natl Agrarian Univ, Fac Engn & Technol, UA-40000 Sumy, Ukraine;[3]Xinxiang Univ, Coll Mech & Elect Engn, Xinxiang 453003, Peoples R China;[4]Chinese Acad Agr Sci, Inst Farmland Irrigat, Xinxiang 453002, Peoples R China

第一机构：Henan Inst Sci & Technol, Sch Informat Engn, Xinxiang 453003, Peoples R China

通讯机构：[1]corresponding author), Henan Inst Sci & Technol, Sch Informat Engn, Xinxiang 453003, Peoples R China.

年份：2023

卷号：23

期号：6

外文期刊名：SENSORS

收录：;EI(收录号：20231613886541);Scopus(收录号：2-s2.0-85151173597);WOS:【SCI-EXPANDED(收录号:WOS:000958893300001)】；

语种：英文

外文关键词：tomato detection; YOLOv5; small-target detection; lightweight

摘要：Due to their rapid development and wide application in modern agriculture, robots, mobile terminals, and intelligent devices have become vital technologies and fundamental research topics for the development of intelligent and precision agriculture. Accurate and efficient target detection technology is required for mobile inspection terminals, picking robots, and intelligent sorting equipment in tomato production and management in plant factories. However, due to the limitations of computer power, storage capacity, and the complexity of the plant factory (PF) environment, the precision of small-target detection for tomatoes in real-world applications is inadequate. Therefore, we propose an improved Small MobileNet YOLOv5 (SM-YOLOv5) detection algorithm and model based on YOLOv5 for target detection by tomato-picking robots in plant factories. Firstly, MobileNetV3-Large was used as the backbone network to make the model structure lightweight and improve its running performance. Secondly, a small-target detection layer was added to improve the accuracy of small-target detection for tomatoes. The constructed PF tomato dataset was used for training. Compared with the YOLOv5 baseline model, the mAP of the improved SM-YOLOv5 model was increased by 1.4%, reaching 98.8%. The model size was only 6.33 MB, which was 42.48% that of YOLOv5, and it required only 7.6 GFLOPs, which was half that required by YOLOv5. The experiment showed that the improved SM-YOLOv5 model had a precision of 97.8% and a recall rate of 96.7%. The model is lightweight and has excellent detection performance, and so it can meet the real-time detection requirements of tomato-picking robots in plant factories.