农业机械学报2024，Vol.55Issue(11)：124-131,374,9.DOI:10.6041/j.issn.1000-1298.2024.11.013
基于图像增强与GC-YOLO v5s的水下环境河蟹识别轻量化模型研究
Lightweight Model for River Crab Detection Based on Image Enhancement and Improved YOLO v5s
摘要
Abstract
Using machine vision technology to identify underwater crab targets is one of the effective ways to achieve intelligent crab farming equipment.However,river crab detection methods face challenges in the difficulty of target detection in underwater environments,limited feature information and high complexity of mainstream target detection models.To solve these challenges,a lightweight river crab detection model GC-YOLO v5s(GhostNetV2-CBAM-YOLO v5s)was proposed.These specific enhancements were as follows:an improved image enhancement algorithm was used to preprocess underwater crab images to improve the detection accuracy;in order to reduce model complexity,a G3 module based on GhostNetV2 was proposed to improve the feature extraction network of the model,and Ghost convolution was used to further lightweight the model;the convolution block attention module(CBAM)was introduced to solve the challenge of extracting deep features within underwater environments,which were integrated into the feature extraction network.The experimental results demonstrated the improved model's mAP50,recall,and precision on the test set,reaching 95.61％,97.03％and 96.94％,respectively.These metrics displayed enhancements of 2.80 percentage points,2.25 percentage points and 2.28 percentage points compared with the baseline.Moreover,GC-YOLO v5s'parameters,computations,and model size were only 69.1％,56.3％,and 58.3％of YOLO v5s respectively.Comparative trials against mainstream object detection algorithms showcased the superiority in accuracy and model complexity.While slightly trailing YOLO v5s in detect speed,GC-YOLO achieved 104 f/s.关键词
水产养殖/河蟹识别模型/图像增强/YOLO v5s/轻量化Key words
aquaculture/river crab detection model/image enhancement/YOLO v5s/lightweight分类
信息技术与安全科学引用本文复制引用
张铮,鲁祥,胡庆松..基于图像增强与GC-YOLO v5s的水下环境河蟹识别轻量化模型研究[J].农业机械学报,2024,55(11):124-131,374,9.基金项目
上海市崇明区农业科创项目(2021CNKC-05-06)和上海市水产动物良种创制与绿色养殖协同创新中心项目(2021科技02-12) （2021CNKC-05-06）
