农业机械学报2024,Vol.55Issue(11):140-146,7.DOI:10.6041/j.issn.1000-1298.2024.11.015
基于改进可变形卷积的FDC-YOLO v8水下生物目标检测方法研究
Research on FDC-YOLO v8 Underwater Biological Object Detection Method Improved by Deformable Convolution
摘要
Abstract
Underwater biological target detection is a crucial technology for achieving automation in underwater robotic fishing.Aiming to address issues such as object overlap,occlusion,and false detections,missed detections caused by small object scales in underwater biological object detection tasks,an underwater biological object detection algorithm,FDC-YOLO v8 was proposed based on an improved YOLO v8n.Firstly,the FDC module was incorporated,which utilized deformable convolution networks in the backbone network to enhance the model's feature extraction capability and enrich the diversity of extracted features.Secondly,the FrSAConv module,integrating fractional Fourier transform and spatial attention mechanism,was introduced to further separate diverse object features and enhance the model's perceptual ability towards various features.Finally,the Wise-IoU loss function was introduced as the bounding box loss function to better address issues related to object imbalance and scale differences.The experiments were conducted by using the RUIE dataset,which included four types of underwater organisms:echinus,starfish,holothurian,and scallops.Experimental results demonstrated that the improved FDC-YOLO v8 achieved an mAP of 85.3%,a 2.6 percentage points improvement over the baseline model.The inference speed can reach 769 frames per second,showcasing better performance in underwater object detection of marine organisms with challenged such as object overlap,occlusion,and small-scale objects.关键词
水下生物识别/目标检测/YOLO v8n/Wise-IoU/可变形卷积网络/分数阶傅里叶变换Key words
underwater biological recognition/object detection/YOLO v8n/Wise-IoU/deformable convolutional network/fractional Fourier transform分类
信息技术与安全科学引用本文复制引用
袁红春,李春桥..基于改进可变形卷积的FDC-YOLO v8水下生物目标检测方法研究[J].农业机械学报,2024,55(11):140-146,7.基金项目
国家自然科学基金项目(41776142) (41776142)