基于YOLOv8的中华鲟智能检测方法
Intelligent Detection Method for Chinese Sturgeon Based on YOLOv8
摘要: 针对水下鱼类视频图像通常存在严重遮挡、近背景色、目标尺寸不一致的问题,在YOLOv8检测算法基础上,引入EMA注意力机制,利用该机制能自动学习并更加准确地捕捉中华鲟关键特征,以提升YOLOv8的预测能力。将引入EMA注意力机制的YOLOv8检测算法用于中华鲟检测的试验结果表明:在精度mAP@0.5方面,改进的YOLOv8模型相比原始YOLOv8模型提高了3.7%,达到了94%。因此引入EMA注意力机制的YOLOv8检测算法能更好地用于水下视频中华鲟检测。
Abstract:
In response to the problems of severe occlusion, near background color, and inconsistent target size in underwater Chinese sturgeon video images, an EMA attention mechanism is introduced based on the YOLOv8 detection algorithm. This mechanism can automatically learn and more accurately capture key features of Chinese sturgeon, improving YOLOv8’s predictive ability. The experimental results of using the YOLOv8 detection algorithm with EMA attention mechanism for detecting Chinese sturgeon show that the accuracy mAP@0.5 in terms of performance, the improved YOLOv8 model has increased by 3.7% compared to the original YOLOv8 model, reaching 94%. Therefore, the YOLOv8 detection algorithm with the introduction of EMA attention mechanism can be better used for underwater video detection of Chinese sturgeon.
参考文献
|
[1]
|
Cao, X., Zhang, X., Yru, Y., et al. (2016) Deep Learning-Based Recognition of Underwater Target. 2016 IEEE Interna-tional Conference on Digital Signal Processing, Beijing, 16-18 October 2016, 89-93. [Google Scholar] [CrossRef]
|
|
[2]
|
袁红春, 张硕. 基于Faster R-CNN和图像增强的水下鱼类目标检测方法[J]. 大连海洋大学学报, 2020, 35(4): 612-619. [Google Scholar] [CrossRef]
|
|
[3]
|
沈军宇, 李林燕, 夏振平, 等. 一种基于YOLO算法的鱼群检测方法[J]. 中国体视学与图像分析, 2018, 90(2): 54-60.
|
|
[4]
|
Santos, A. and Gonalves, W.N. (2019) Improving Pantanal Fish Species Recognition through Taxonomic Ranks in Convolutional Neural Networks. Ecological Informatics, 53, 100977-100977. [Google Scholar] [CrossRef]
|
|
[5]
|
Zhang, S., Yang, X., Wang, Y., et al. (2020) Automatic Fish Population Counting by Machine Vision and a Hybrid Deep Neural Network Model. Animals, 10, Article 364. [Google Scholar] [CrossRef] [PubMed]
|
|
[6]
|
李辉, 王晓宇, 刘云, 等. 融合多尺度特征和多重注意力的水下目标检测[J]. 农业工程学报, 2022, 38(20): 129-139.
|
|
[7]
|
周温晖, 贾永红, 焦阳. 水下视频中华鲟检测方法研究[J]. 计算机科学与应用, 2022, 12(8): 1998-2005
|
|
[8]
|
施克权, 李祺, 隋皓, 等. IEMAyoloViT: 基于改进YOLOv8的水下目标检测算法[J/OL]. 电讯技术, 2024: 1-10.
2024-04-16.[CrossRef]
|