Identification of Pine Wilt Disease Infected Wood Using UAV RGB Imagery and Improved YOLOv5 Models Integrated with Attention Mechanisms

文献类型： 外文期刊

第一作者： Zhang, Peng

作者： Zhang, Peng;Zhu, Jianqiao;Fang, Yifan;Gao, Xiang;Zhang, Peng;Rao, Yuan;Wang, Zhichao;Zheng, Jun;Zhang, Ning;Wang, Degao

作者机构：

关键词： pine wilt disease; YOLOv5; diseased wood; UAV remote sensing; attention mechanism

期刊名称：FORESTS （ 影响因子：2.9； 五年影响因子：3.0 ）

ISSN：

年卷期： 2023 年 14 卷 3 期

页码：

收录情况： SCI

摘要： Pine wilt disease (PWD) is a great danger, due to two aspects: no effective cure and fast dissemination. One key to the prevention and treatment of pine wilt disease is the early detection of infected wood. Subsequently, appropriate treatment can be applied to limit the further spread of pine wilt disease. In this work, a UAV (Unmanned Aerial Vehicle) with a RGB (Red, Green, Blue) camera was employed as it provided high-quality images of pine trees in a timely manner. Seven flights were performed above seven sample plots in northwestern Beijing, China. Then, raw images captured by the UAV were further pre-processed, classified, annotated, and formed the research datasets. In the formal analysis, improved YOLOv5 frameworks that integrated four attention mechanism modules, i.e., SE (Squeeze-and-Excitation), CA (Coordinate Attention), ECA (Efficient Channel Attention), and CBAM (Convolutional Block Attention Module), were developed. Each of them had been shown to improve the overall identification rate of infected trees at different ranges. The CA module was found to have the best performance, with an accuracy of 92.6%, a 3.3% improvement over the original YOLOv5s model. Meanwhile, the recognition speed was improved by 20 frames/second compared to the original YOLOv5s model. The comprehensive performance could well support the need for rapid detection of pine wilt disease. The overall framework proposed by this work shows a fast response to the spread of PWD. In addition, it requires a small amount of financial resources, which determines the duplication of this method for forestry operators.

分类号：

  • 相关文献
作者其他论文 更多>>

微信小程序