基于YOLO-Dense的温室场景下番茄异常检测

Tomato Anomalies Detection in Greenhouse Scenarios Based on YOLO-Dense.

作者信息

Wang Xuewei, Liu Jun

机构信息

Shandong Provincial University Laboratory for Protected Horticulture, Blockchain Laboratory of Agricultural Vegetables, Weifang University of Science and Technology, Weifang, China.

出版信息

Front Plant Sci. 2021 Apr 9;12:634103. doi: 10.3389/fpls.2021.634103. eCollection 2021.

Abstract

Greenhouse cultivation can improve crop yield and quality, and it not only solves people's daily needs but also brings considerable gains to the agricultural staff. One of the most widely cultivated greenhouse crops is tomato, mainly because of its high nutritional value and its good taste. However, there are a number of anomalies for the tomato crop that pose a threat for its greenhouse cultivation. Detection of tomato anomalies in the complex natural environment is an important research direction in the field of plant science. Automated identification of tomato anomalies is still a challenging task because of its small size and complex background. To solve the problem of tomato anomaly detection in the complex natural environment, a novel YOLO-Dense was proposed based on a one-stage deep detection YOLO framework. By adding a dense connection module in the network architecture, the network inference speed of the proposed model can be effectively improved. By using the K-means algorithm to cluster the anchor box, nine different sizes of anchor boxes with potential objects to be identified were obtained. The multiscale training strategy was adopted to improve the recognition accuracy of objects at different scales. The experimental results show that the mAP and detection time of a single image of the YOLO-Dense network is 96.41% and 20.28 ms, respectively. Compared with SSD, Faster R-CNN, and the original YOLOv3 network, the YOLO-Dense model achieved the best performance in tomato anomaly detection under a complex natural environment.

摘要

温室栽培可以提高作物产量和品质,它不仅解决了人们的日常需求,也给农业工作者带来了可观的收益。温室中种植最广泛的作物之一是番茄,主要是因为它营养价值高且口感好。然而,番茄作物存在一些异常情况,对其温室栽培构成威胁。在复杂自然环境中检测番茄异常是植物科学领域的一个重要研究方向。由于番茄体积小且背景复杂,自动识别番茄异常仍然是一项具有挑战性的任务。为了解决复杂自然环境下番茄异常检测的问题,基于单阶段深度检测YOLO框架提出了一种新颖的YOLO-Dense。通过在网络架构中添加密集连接模块,可以有效提高所提模型的网络推理速度。利用K均值算法对锚框进行聚类,得到了九种不同大小的带有潜在待识别物体的锚框。采用多尺度训练策略提高不同尺度物体的识别精度。实验结果表明,YOLO-Dense网络单张图像的平均精度均值(mAP)和检测时间分别为96.41%和20.28毫秒。与SSD、Faster R-CNN和原始的YOLOv3网络相比,YOLO-Dense模型在复杂自然环境下的番茄异常检测中表现最佳。

https://cdn.ncbi.nlm.nih.gov/pmc/blobs/9cd4/8063041/a951b6c28108/fpls-12-634103-g001.jpg

文献AI研究员

20分钟写一篇综述,助力文献阅读效率提升50倍

立即体验

用中文搜PubMed

大模型驱动的PubMed中文搜索引擎

马上搜索