标题
  • 标题
  • 作者
  • 关键词

基于改进YOLOv5_OBB的中华绒螯蟹旋转目标检测

2024-03-18分类号:TP391.41;S966.16

【作者】袁红春   白宝来   陶磊
【部门】上海海洋大学信息学院  
【摘要】【目的】提出一种基于改进YOLOv5_OBB的旋转目标检测方法,快速、准确地检测和定位中华绒螯蟹。【方法】首先,在YOLOv5_OBB的主干网络中引入高效通道注意模块;其次,采用BiFPN网络结构进行特征融合模块设计,实现高效的双向跨尺度连接和加权特征融合;最后,采用变焦损失(varifocal loss)解决正负样本不均衡问题。【结果】改进后YOLOv5_OBB模型的P(precision)、R(recall)和mAP(mean average precision)分别达到95.4%、95.2%和90.1%,比原模型分别提高了1.0%、1.9%和1.3%。【结论】该模型能够实时、准确地检测和定位中华绒螯蟹,实现自动化养殖。
【关键词】中华绒螯蟹  YOLOv5_OBB  旋转目标检测
【基金】国家自然科学基金项目(41776142)
【所属期刊栏目】福建农林大学学报(自然科学版)
文献传递