《一种基于轻量化YOLO v5的甘蔗蚜虫目标检测方法》

技术简介：

本发明涉及目标检测技术领域,尤其涉及一种基于轻量化YOLO v5的甘蔗蚜虫目标检测方法,包括：S1：甘蔗蚜虫图像采集；S2：图像预处理；S3：制作数据集；S4：对YOLO v5模型轻量化改进；S5：输入数据集训练得到轻量化模型。本发明使用Stem模块,ShuffleNetV2的Inverted Residual模块,减少颈部网络宽度来对YOLO v5模型轻量化,同时删除大尺度检测层,增加微小尺度检测层,使得网络能够更好地符合甘蔗蚜虫中密集微小目标的检测,与原始YOLO v5模型相比,具有识别精度高、识别速度快、内存占用率低等优点,有利于模型在移动终端上部署。

研发人员：徐伟悦;徐涛;王炎;陈伟;张储