首页|Tree Detection Algorithm Based on Embedded YOLO Lightweight Network

Tree Detection Algorithm Based on Embedded YOLO Lightweight Network

扫码查看
To avoid colliding with trees during its operation,a lawn mower robot must detect the trees.Existing tree detection methods suffer from low detection accuracy(missed detection)and the lack of a lightweight model.In this study,a dataset of trees was constructed on the basis of a real lawn environment.According to the theory of channel incremental depthwise convolution and residual suppression,the Embedded-A module is proposed,which expands the depth of the feature map twice to form a residual structure to improve the lightweight degree of the model.According to residual fusion theory,the Embedded-B module is proposed,which improves the accuracy of feature-map downsampling by depthwise convolution and pooling fusion.The Embedded YOLO object detection network is formed by stacking the embedded modules and the fusion of feature maps of different resolutions.Experimental results on the testing set show that the Embedded YOLO tree detection algorithm has 84.17%and 69.91%average precision values respectively for trunk and spherical tree,and 77.04%mean average precision value.The number of convolution parameters is 1.78 x 106,and the calculation amount is 3.85 billion float operations per second.The size of weight file is 7.11MB,and the detection speed can reach 179 frame/s.This study provides a theoretical basis for the lightweight application of the object detection algorithm based on deep learning for lawn mower robots.

Embedded YOLO algorithmlightweight modelmachine visiontree detectionmowing robot

吕峰、王新彦、李磊、江泉、易政洋

展开 >

School of Mechanical Engineering,Jiangsu University of Science and Technology,Zhenjiang 212100,Jiangsu,China

National Natural Science Foundation of China

51275223

2024

上海交通大学学报(英文版)
上海交通大学

上海交通大学学报(英文版)

影响因子:0.151
ISSN:1007-1172
年,卷(期):2024.29(3)