具有双层路由注意力机制的YOLOv8血鹦鹉目标检测与追踪方法
YOLOv8 blood parrot object detection and tracking method with dual-layer routing attention mechanism
李鹏龙 1张胜茂 2沈烈 3樊伟 2顾家辉 3邹国华4
作者信息
- 1. 大连海洋大学航海与船舶工程学院,辽宁大连 116023;中国水产科学研究院东海水产研究所农业农村部渔业遥感重点试验室,上海 200090
- 2. 中国水产科学研究院东海水产研究所农业农村部渔业遥感重点试验室,上海 200090
- 3. 大连海洋大学航海与船舶工程学院,辽宁大连 116023
- 4. 上海峻鼎渔业科技有限公司,上海 200090
- 折叠
摘要
为了检测观赏鱼类的行为及其健康状况,设计了一种具有双层路由注意力机制的血鹦鹉(Vieja syn-spila ♀ ×Amphilophus citrinellus ♂)目标检测模型 YOLOv8n-BiFormer,该方法在 YOLOv8n 模型基础上添加了双层路由注意力以减少计算量和内存,添加了新的视觉通用变换器BiFormer以提升计算效率,并采用ByteTrack算法追踪血鹦鹉的运动轨迹.结果表明:使用YOLOv8n-BiFormer模型对血鹦鹉的检测准确率达到99.2％,召回率为93.7％,平均精度均值(mAP@0.5)为99.1％,相较于YOLOv8n模型分别提升了0.8％、1.4％、1.0％;使用该模型对水族箱中的慈鲷(Chindongo demasoni)进行检测追踪同样取得了较好的效果,慈鯛的检测准确率达到97.0％,召回率为93.4％,平均精度均值为96.5％,相较于YOLOv8n模型召回率和平均精度分别提升了 1.8％和1.9％.研究表明,本文中设计的YOLOv8n-BiFormer模型具有通用性,在检测和追踪血鹦鹉和慈鲷目标方面均表现优异,消耗的计算资源较少,可部署在水族箱监控系统中,为观赏鱼信息记录自动化和智能化提供了可行的解决方案.
Abstract
In order to detect the behavior and health status of hybrid ornamental fish(Vieja synspila ♀×Amphilo-phus citrinellus ♂),a target detection model called YOLOv8n-BiFormer with a dual-layer routing attention mecha-nism was designed.In this method a dual-layer routing attention mechanism is added into the YOLOv8n model to reduce computation and memory requirements,and a new visual universal transformer called BiFormer is introduced to the YOLOv8n model for improvement of computational efficiency.The ByteTrack algorithm is employed to track the motion trajectory of the fish blood parrot.The results showed that the YOLOv8n-BiFormer model had a detection accuracy of 99.2％,a recall rate of 93.7％,and an average precision of 99.1％(mAP@0.5)for the blood par-rot,increased by 0.8％,1.4％,and 1.0％compared to the YOLOv8n model,respectively.The model demon-strated good performance in the detection and tracking of the cichlid(Chindongo demasoni)in an aquarium,with a detection accuracy of 97.0％,a recall rate of 93.4％,and an average precision of 96.5％,increase by 1.8％in recall rate and 1.9％in average precision compared to the YOLOv8n model.The finding demonstrates that the de-signed YOLOv8n-BiFormer model performs excellently in detecting and tracking blood parrot fish and cichlid tar-gets,with fewer computational resources,and that can be deployed in aquarium monitoring systems,providing fea-sible solution for the automation and intelligence of ornamental fish information recording.
关键词
血鹦鹉/慈鲷/YOLOv8模型/检测追踪/ByteTrack算法Key words
Vieja synspila ♀×Amphilophus citrinellus ♂/Chindongo demasoni/YOLOv8 model/detection and tracking/ByteTrack algorithm引用本文复制引用
基金项目
国家自然科学基金(61936014)
崂山实验室专项(LSKJ202201804)
出版年
2024
NETL
NSTL
维普
万方数据