农业工程学报2023，Vol.39Issue(17)：163-171,9.DOI:10.11975/j.issn.1002-6819.202306084
基于改进YOLOv5的菇房平菇目标检测与分类研究
Object detection and classification of pleurotus ostreatus using improved YOLOv5
摘要
Abstract
Accurate detection of pleurotus ostreatus is highly required in modern mushroom houses,particularly with the ever-increasing automation of intelligence and informatization.However,manual harvesting cannot fully meet the large-scale factory cultivation in the edible mushroom industry.Fortunately,a harvesting robot can be expected to increase productivity with the reduced labor intensity at the industrial cultivation and harvesting stage.In this article,an improved OMM-YOLO(ostreatus measure modle-you only look once)target detection model was proposed to conduct comparative experiments on the mushroom detection dataset.1 353 images of mushrooms were collected from the mushroom houses and then classified into the mature,immature,and ungrown at the growth stages.An OMM-YOLO target detection and classification model was proposed using the improved YOLOv5 model.The EC A(effective channel attention)attention module was also introduced into the Backbone layer of the original YOLOv5 model.The features of input mushroom images were dynamically weighted to obtain more feature information.At the same time,a weighted bidirectional feature pyramid network(BiFPN)was used in the Neck layer.Multi-scale features were repeatedly fused from the top to the bottom and then bottom-up,thereby improving the accuracy of mushroom target detection.In addition,the EIoU(enhanced intersection over union)Loss function was utilized to improve the accuracy of the model and the rate of convergence of the bounding box aspect ratio,instead of the original Loss function CIoU(Complete IoU).The better performance was achieved in terms of the average accuracy mAP,and the rate of convergence of the bounding box aspect ratio.The experimental results show that the improved OMM-YOLO model shared an average accuracy mAP of 91.4％,87.1％,and 95.1％for the mature,immature,and ungrown mushrooms,respectively,which was improved by 0.4,4.5,and 1.1 percentage points,respectively,compared with the original.Better performance was achieved in the improved OMM-YOLO model,in terms of the accuracy,recall,detection accuracy,and detection speed,compared with the current mainstream models,such as Resnet50,VGG16,YOLOv3,YOLOv4,YOLOv5m,and YOLOv7.Therefore,this improved model was very suitable for the detection of mushrooms in modem mushroom houses.Mushroom features were collected to effectively avoid the occurrence of false detection caused by the mutual occlusion of mushrooms.An attention mechanism was introduced to weigh the BiFPN using a more appropriate Loss function.The accuracy and rate of convergence of the improved model were significantly enhanced for the mushroom target detection.A more effective technical means can be expected in the detection of mushroom targets in modem mushroom houses at present.As such,mushroom target detection can automatically detect the quantity and growth condition of the mushroom in the mushroom room.Environmental parameters(such as temperature and humidity)can be adjusted in a timely manner.The production efficiency can be promoted to control the quality of the mushroom for better uniformity and quality stability of the products.At the same time,it can be used to reduce the dependence on labor and costs.Sustainable development can be realized through the positive impact of the construction of intelligent mushroom houses.关键词
目标检测/分类/模型/高效通道注意力模块/平菇/加权双向特征金字塔/EIoU损失函数Key words
target detection/classification/models/efficient channel attention module/pleurotus ostreatus/weighted bidirectional feature pyramid/EIoU loss function分类
信息技术与安全科学引用本文复制引用
王磊磊,王斌,李东晓,赵义鹏,王春霞,张迪迪..基于改进YOLOv5的菇房平菇目标检测与分类研究[J].农业工程学报,2023,39(17):163-171,9.基金项目
河北省农业科技成果转化项目(202360101010014) （202360101010014）
河北省高等学校学科技术研究计划(QN2021210). （QN2021210）
