RepDI: A light-weight CPU network for apple leaf disease identification

文献类型： 外文期刊

第一作者： Zheng, Jiye

作者： Zheng, Jiye;Li, Kaiyu;Ruan, Huaijun;Li, Kaiyu;Wu, Wenbin

作者机构：

关键词： Disease identification; Image processing; Deep learning; Lightweight model

期刊名称：COMPUTERS AND ELECTRONICS IN AGRICULTURE （ 影响因子：8.3； 五年影响因子：8.3 ）

ISSN： 0168-1699

年卷期： 2023 年 212 卷

页码：

收录情况： SCI

摘要： Apple disease is one of the major factors affecting apple production, and the visual diagnosis of apple leaves is an efficient disease identification solution. In this paper, we propose an efficient lightweight model based on structural reparameterization for apple leaf disease identification, called RepDI for short. To achieve faster inference on the CPU devices, we introduce depth-wise separable convolution and structural reparameterization technology in RepDI, which has different structures during training and inference. In addition, to better capture diseased leaves and disease regions in complex contexts, we propose the parallel dilated attention mechanism module and embed it into RepDI. Experiments show that RepDI can achieve state-of-the-art performance in disease identification task, compared to most lightweight models. Meanwhile, RepDI achieves the fastest inference speed on our desktop CPU, which is an important factor in practical applications. Furthermore, we collect and annotate a novel dataset for apple leaf diseases from real scenarios, called Real-ALD, which is more challenging than previous datasets. And RepDI achieves a top-1 accuracy of 98.92 in the Real-ALD dataset under a limited training configuration. Our code is released to contribute to the plant protection community and we will further explore the potential of RepDI for down-stream detection, segmentation tasks.

分类号：

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

微信小程序