论文

改进YOLO v4模型在鱼类目标检测上的应用研究

  • 郑宗生 ,
  • 李云飞 ,
  • 卢鹏 ,
  • 邹国良 ,
  • 王振华
展开
  • (上海海洋大学信息学院,上海 201306)
郑宗生(1979—),男,副教授,博士,研究方向:深度学习、遥感图像处理。E-mail: zgsong@shou.edu.cn

网络出版日期: 2022-07-06

基金资助

上海市科委地方院校能力建设项目(190505022100);国家海洋局科学技术重点实验室开放基金项目(B201801034);科技发展专项基金(A2-2006-20-200211)

Application research of improved YOLO v4 model in fish object detection

  • ZHENG Zongsheng ,
  • LI Yunfei ,
  • LU Peng ,
  • ZOU Guoliang ,
  • WANG Zhenhua
Expand
  • (Shanghai Ocean University, Colleage of Information, Shanghai 201306, China)

Online published: 2022-07-06

摘要

鱼类目标检测对渔业精准养殖、生产自动化、资源调查及鱼行为的研究等具有重要的意义。为了能快速准确地得到鱼类目标的位置和所属类别,提出了一种改进YOLO v4模型的鱼类目标检测方法,在CIoU(Complete Intersection over Union)损失函数基础上构建了新的损失项,改进的损失函数使真实框与相交框呈相同宽高比进行回归,同时通过设置多锚点框模式,增强在特定尺寸面积上的检测效果。结果显示:改进YOLO v4模型的mAP(mean Average Precision)比原模型有较大提升,在自建数据集、Fish4Knowledge数据集和NCFM数据集上的mAP分别达到了94.22%、99.52%、92.16%。研究表明,改进YOLO v4模型可以快速准确地检测到鱼的位置和类别,检测速度满足实时的要求,可以为渔业精准养殖等提供参考。

本文引用格式

郑宗生 , 李云飞 , 卢鹏 , 邹国良 , 王振华 . 改进YOLO v4模型在鱼类目标检测上的应用研究[J]. 渔业现代化, 2022 , 49(1) : 82 -88 . DOI: 10.3969/j.issn.1007-9580.2022.01.011

Abstract

Fish object detection is of great significance for precision aquaculture, production automation, resource investigation and fish behavior research. In order to get the position and category of fish object quickly and accurately, a fish object detection method based on improved model YOLO V4 is proposed, Based on the CIoU(complete intersection over union) loss function, a new loss term is constructed. The improved loss function makes the real box and the intersecting box regress in the same aspect ratio. At the same time, the detection effect on a specific size and area is enhanced by setting a multi anchor box mode. The results show that the mAP (mean average precision) of the improved model YOLO V4 is greatly improved compared with the original model. The mAP on the self built data set, data set Fish4 knowledge and data set NCFM reaches 94.22%, 99.52% and 92.16% respectively. The research shows that the improved model YOLO v4 can quickly and accurately detect the position and category of fish, and the detection speed meets the real-time requirements, which can provide a reference for precision aquaculture of fishery.
文章导航

/