渔业现代化2024，Vol.51Issue(3)：89-97,9.DOI:10.3969/j.issn.1007-9580.2024.03.010
基于改进YOLOv5n的轻量化海产生物目标检测
Lightweight marine organism target detection based on improved YOLOv5
摘要
Abstract
In marine biology research and fishing industries,ships using underwater robots to capture and identify marine organisms may face limited communication bandwidth and computing resources,making lightweight network models better suited for such conditions.To address this issue,a modified YOLOv5n model for marine organism detection was proposed,incorporating improvements in the neck network,model pruning,and knowledge distillation techniques.The model size was effectively reduced by leveraging the Gsconv lightweight convolution module to replace standard convolutions in the YOLOv5n Neck section.A novel α-giou loss function was adopted to enhance bounding box regression accuracy.Based on weight coefficients,L1-norm regularization pruning was applied to eliminate unnecessary channels and associated convolutional kernels.Finally,retraining and L2 knowledge distillation were employed to fine-tune the model accuracy close to pre-pruning levels.Experimental results demonstrated a 53％reduction in computational load and a 51％decrease in parameters compared to the original YOLOv5n baseline network.The proposed algorithm ensures the effectiveness.关键词
海产生物目标检测/YOLOv5n/Gsconv/模型剪枝/知识蒸馏Key words
marine biological target detection/YOLOv5n/gsconv/model pruning/knowledge distillation分类
海洋科学引用本文复制引用
张翔,张俊虎,李海涛,李辉..基于改进YOLOv5n的轻量化海产生物目标检测[J].渔业现代化,2024,51(3):89-97,9.基金项目
国家自然科学基金(61702295) （61702295）
青岛市海洋科技创新专项(22-3-3-hygg-hy) （22-3-3-hygg-hy）
