###
计算机系统应用英文版:2023,32(2):94-101
本文二维码信息
码上扫一扫!
基于TensorRT的植物叶片病害实时检测分类模型优化
(1.五邑大学 智能制造学部, 江门 529020;2.广东邮电职业技术学院 计算机学院, 广州 510630)
Optimization of Real-time Detection and Classification Model for Plant Leaf Diseases Based on TensorRT
(1.Faculty of Intelligent Manufacturing, Wuyi University, Jiangmen 529020, China;2.School of Computer, Guangdong Vocational College of Post and Telecom, Guangzhou 510630, China)
摘要
图/表
参考文献
相似文献
本文已被:浏览 654次   下载 1971
Received:June 20, 2022    Revised:September 07, 2022
中文摘要: 为了提高边缘计算设备对植物叶片病害检测的识别速率, 本研究采用卷积神经网络搭建了植物叶片目标识别模型和植物叶片病害分类模型, 并且使用OpenCV将两个模型整合成植物叶片病害检测系统. 通过SSD (single shot multibox detector)算法对植物叶片的目标区域进行定位并裁剪, 再利用植物叶片病害分类模型对裁剪的植物叶片区域进行病害分类. 同时, 通过TensorRT加速推理对分类模型进行优化处理, 以及在同一台主机设备和Jetson Nano计算平台上, 对优化前后的模型进行了对比实验. 实验表明, 在同一主机设备上优化后的植物分类模型识别速率提升22倍. 同时, 优化后的分类模型使植物叶片病害检测系统识别速率提升7倍. 而将优化后的系统部署在Jetson Nano计算平台上, 对比优化前的植物叶片病害检测速率提升10倍, 实现了实时的植物叶片病害检测.
Abstract:In order to improve the recognition rate of plant leaf disease detection by edge computing devices, this study uses a convolutional neural network to build a plant leaf target recognition model and a plant leaf disease classification model and adopts OpenCV to integrate the two models into a plant leaf disease detection system. The target areas of plant leaves are positioned and clipped by the single shot multibox detector (SSD) algorithm, and then the plant leaf disease classification model is used to classify the clipped plant leaf areas according to diseases. At the same time, the classification model is optimized by TensorRT accelerated inference. In addition, on the same host device and Jetson Nano computing platform, a comparative experiment is carried out on the model before and after optimization. The experiment shows that the recognition rate of the optimized plant classification model on the same host device increases by 22 times. At the same time, the optimized classification model improves the recognition rate of the plant leaf disease detection system by seven times. Furthermore, the optimized system is deployed on the Jetson Nano computing platform, and the detection rate of plant leaf diseases is increased by 10 times compared with that before optimization, which thus realizes real-time plant leaf disease detection.
文章编号:     中图分类号:    文献标志码:
基金项目:广东邮电职业技术学院校级质量工程项目(202201); 广东省职业技术教育学会第四届理事会科研规划项目(202103G93); 2022年度广东省普通高校特色创新类项目(2022KTSCX288)
引用文本:
徐泽华,李坚孝,邓树源,吴家隐,高嘉晖,潘明毅.基于TensorRT的植物叶片病害实时检测分类模型优化.计算机系统应用,2023,32(2):94-101
XU Ze-Hua,LI Jian-Xiao,DENG Shu-Yuan,WU Jia-Yin,GAO Jia-Hui,PAN Ming-Yi.Optimization of Real-time Detection and Classification Model for Plant Leaf Diseases Based on TensorRT.COMPUTER SYSTEMS APPLICATIONS,2023,32(2):94-101