基于改进YOLOv8的轻量级鱼类检测方法OA北大核心CSTPCD
Lightweight fish detection method based on improved YOLOv8
针对鱼类目标检测存在精度低和计算量大的问题,提出了一种基于改进YOLOv8 模型的轻量化鱼类目标检测方法YOLOv8-FCW.首先,引入FasterNet中的FasterBlock替换YOLOv8 中C2f模块的Bottleneck结构,减少网络模型的冗余计算;其次,引入注意力机制CBAM(Convolutional Block Attention Module),实现高效提取鱼体特征,提升网络模型检测精度;最后,引入动态非单调聚焦机制WIoU(Wise Intersection over Union)替代CIoU(Complete Intersection over Union),加快网络模型的收敛速度,提升网络模型的检测性能.结果显示,与原模型相比,改进YOLOv8-FCW模型精确率提升了 1.6 个百分点,召回率提升了 5.1 个百分点,平均精确率均值提升了 2.4 个百分点,权重和计算量分别减少为原模型的 80%和 79%.该模型具有较高的精确率和较强的鲁棒性,能够帮助养殖者精确计算鱼群数量,提高养殖效率.
Fish culture is moving toward precision culture,and fish target detection is an important part of precision culture.Fortunately,the use of deep learning holds promise for fish target detection.However,the existing fish target detection models have problems with heavy computation and low accuracy.To address the issues of low accuracy and high computational load in fish target detection,a lightweight fish target detection method based on an improved YOLOv8 model was proposed and named YOLOv8-FCW in this study.Firstly,The experimental comparison of MobileNet,ShuffleNet,GhostNet and C2f-Faster shows that C2f-Faster performs best.Therefore,the FasterBlock from FasterNet was introduced to replace the Bottleneck module in C2f of YOLOv8,reducing redundant computations in the network model.Secondly,the Convolutional Block Attention Module(CBAM)attention mechanism was incorporated to efficiently extract fish body features and enhance the detection accuracy of the network model.Finally,The experimental results show that the loss value and convergence speed of the Wise intersection over union(WIoU)loss function is better than Complete intersection over union(CIoU),Distance intersection over union(DIoU)and Generalized intersection over union(GIoU).Therefore,a dynamic non-monotonic focusing mechanism WIoU was introduced to replace CIoU,accelerating the convergence speed of the network model and improving its detection performance.To verify the detection effect of YOLOv8-FCW on fish,the original model and YOLOv8-FCW were trained and tested on the fish data set.The fish data set consists of 1000 images,which were divided into training set,verification set and test set according to the ratio of 8∶1∶1.Experimental results show that compared with the original model,the improved YOLOv8-FCW model had increased precision by 1.6 percentage points,recall by 5.1 percentage points,and mean average precision(mAP)by 2.4 percentage points,while the weight and computational load were reduced to 80%and 79%of the original model,respectively.YOLOv8-FCW achieves high detection accuracy and efficiency with very small model volume and low computational cost.The model shows high accuracy and robustness.The research can help breeders accurately calculate the number of fish and provide technical references for fish target detection.
王鑫怡;刘旭腾;郑纪业;董贯仓;于兆慧;张霞;王兴家
山东省农业科学院农业信息与经济研究所,山东 济南 250100||聊城大学物理科学与信息工程学院,山东 聊城 252000香港中文大学(深圳),广东 深圳 518172山东省淡水渔业研究院,山东 济南 250013山东东润仪表科技股份有限公司,山东 烟台 264003聊城大学物理科学与信息工程学院,山东 聊城 252000
水产学
图像处理图像识别目标检测YOLOv8
image processingimage recognitiontarget detectionYOLOv8
《渔业现代化》 2024 (006)
91-99 / 9
山东省重点研发计划项目"海洋渔业智能装备与大数据平台系统开发及应用(2021TZXD006)"
评论