期刊
AGRONOMY-BASEL
卷 13, 期 11, 页码 -出版社
MDPI
DOI: 10.3390/agronomy13112667
关键词
computer vision; deep learning; attention mechanism; citrus detection; YOLOv7-tiny
This paper proposes a citrus detection model YOLO-DCA suitable for complex citrus orchard environments, which has high detection accuracy and small model space occupation. It achieves good detection effect in occlusion, light transformation, and motion change scenes, and can help improve the intelligence level of citrus-picking robots.
In complex citrus orchard environments, light changes, branch shading, and fruit overlapping impact citrus detection accuracy. This paper proposes the citrus detection model YOLO-DCA in complex environments based on the YOLOv7-tiny model. We used depth-separable convolution (DWConv) to replace the ordinary convolution in ELAN, which reduces the number of parameters of the model; we embedded coordinate attention (CA) into the convolution to make it a coordinate attention convolution (CAConv) to replace the ordinary convolution of the neck network convolution; and we used a dynamic detection head to replace the original detection head. We trained and evaluated the test model using a homemade citrus dataset. The model size is 4.5 MB, the number of parameters is 2.1 M, mAP is 96.98%, and the detection time of a single image is 5.9 ms, which is higher than in similar models. In the application test, it has a better detection effect on citrus in occlusion, light transformation, and motion change scenes. The model has the advantages of high detection accuracy, small model space occupation, easy application deployment, and strong robustness, which can help citrus-picking robots and improve their intelligence level.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据