吉林大学学报(地球科学版) ›› 2025, Vol. 55 ›› Issue (1): 328-339.doi: 10.13278/j.cnki.jjuese.20240305
杨笑天1, 2,谭金林1, 2,鱼昕1, 2,赵俊哲3,刘铭3
Yang Xiaotian1,2, Tan Jinlin1,2, Yu Xin1,2, Zhao Junzhe3, Liu Ming3
摘要: 针对卫星遥感图像舰船目标追踪和轨迹绘制,提出了一种全局注意力(global attention mechanism, GAM)模块改进YOLO(you only look once)v8算法(GAM-YOLOv8)与DeepSORT算法相结合的方法。在YOLOv8网络结构中加入GAM模块,以提升模型提取卫星遥感图像特征的能力,提高舰船目标追踪的精度和稳定性;实施基于RGB(red, green, blue)-HSV(hue, saturation, value)融合颜色空间转换卷积模块的数据增强技术扩充数据集,帮助模型捕捉更多维度的特征信息,进一步提升识别准确度;利用DeepSORT算法通过结合目标的特征外观和运动信息,增强追踪过程中的稳定性与精度,有效减少身份切换和目标丢失。实验结果表明,本文提出的GAM-YOLOv8与DeepSORT相结合的方法,相较于原始YOLOv8模型,在遥感图像舰船目标检测与跟踪任务中均表现出了显著的性能提升,在准确度、召回率和平均准确率精度上分别提高了7.6%、7.9%和6.0%,在帧率、多目标跟踪准确度和多目标跟踪精确度上分别提升了17.7%、6.9%、1.9%,身份切换次数降低了10.0%。
中图分类号:
| [1] | 高康哲, 王凤艳, 刘子维, 王明常. 基于改进U-Net的遥感图像语义分割[J]. 吉林大学学报(地球科学版), 2024, 54(5): 1752-1763. |
| [2] | 王婷婷, 孙振轩, 戴金龙, 姜基露, 赵万春. 松辽盆地中央坳陷区储层岩性智能识别方法 [J]. 吉林大学学报(地球科学版), 2023, 53(5): 1611-1622. |
| [3] | 刘 霞, 孙英杰. 基于融合残差注意力机制的卷积神经网络地震信号去噪[J]. 吉林大学学报(地球科学版), 2023, 53(2): 609-. |
| [4] | 张岩, 刘小秋, 李杰, 董宏丽, . 基于时频联合深度学习的地震数据重建[J]. 吉林大学学报(地球科学版), 2023, 53(1): 283-296. |
| [5] | 熊越晗, 刘东燕, 刘东升, 王艳磊, 唐小山. 基于岩样细观图像深度学习的岩性自动分类方法[J]. 吉林大学学报(地球科学版), 2021, 51(5): 1597-1604. |
| [6] | 王新民, 张超超. 基于深度学习的旧金山湾水质预测[J]. 吉林大学学报(地球科学版), 2021, 51(1): 222-230. |
| [7] | 代丽艳, 董宏丽, 李学贵. 微地震数据去噪方法综述[J]. 吉林大学学报(地球科学版), 2019, 49(4): 1145-1159. |
|