科研之友
微信
新浪微博
Facebook
分享链接
摘要
由于环境和光线等条件影响，虫害图像识别难度较大，为此提出改进YOLOv5模型下农作物虫害图像识别算法。通过K-means算法改进并优化暗通道图像去雾算法，利用优化后的暗通道图像去雾算法对农作物虫害图像展开处理，以此避免因水汽或光照不均等因素造成的图像清晰度低的问题；对原始YOLOv5模型中的注意力机制和损失函数展开优化，通过优化注意力机制提高农作物虫害图像特征提取的精度，根据优化后的损失函数调整预测框的位置以此锁定虫害目标；将去雾后的农作物虫害图像输入改进后的YOLOv5模型中，实现农作物虫害图像的识别。仿真结果表明，经过所提算法改进后的YOLOv5模型IoU有所提高，且特征提取精度和图像识别效率高。
