公开(公告)号：CN118015049B

公开(公告)日：2024-06-21

申请号：CN202410422431.0

申请日：2024-04-09

Applicant: 湖南大学

Inventor： 陈华 ,  梁凉 ,  张小刚

IPC: G06T7/246 ,  G06N3/0442 ,  G06N3/0464 ,  G06N3/082 ,  G06N3/084 ,  G06V10/25 ,  G06V10/42 ,  G06V10/44 ,  G06V10/764 ,  G06V10/82 ,  G06V10/86

Abstract: 一种结合YOLO和GCN的海上目标追踪方法及其模型，该方法包括：对海上船只数据进行实地采集；实验数据预处理，将训练集输入到检测模块，对图像目标进行特征提取，得到多帧目标识别数据，并计算检测模块的损失；反向传播进行检测模块的YOLO参数更新；将检测模块得到的多帧目标识别数据输入到关联模块，以检测目标为节点构建图结构，提取高级目标特征，进行下一时刻目标位置的预测，结果计算总的损失；通过梯度反向传播进行关联模块参数的更新，得到训练好的海上目标追踪模型；将测试集输入训练好的海上目标追踪模型，输出检测追踪结果和概率。本发明考虑到不同目标之间的空间关系，目标追踪性能好。

公开(公告)号：CN118015049A

公开(公告)日：2024-05-10

申请号：CN202410422431.0

申请日：2024-04-09

Applicant: 湖南大学

Inventor： 陈华 ,  梁凉 ,  张小刚

IPC: G06T7/246 ,  G06N3/0442 ,  G06N3/0464 ,  G06N3/082 ,  G06N3/084 ,  G06V10/25 ,  G06V10/42 ,  G06V10/44 ,  G06V10/764 ,  G06V10/82 ,  G06V10/86

Abstract: 一种结合YOLO和GCN的海上目标追踪方法及其模型，该方法包括：对海上船只数据进行实地采集；实验数据预处理，将训练集输入到检测模块，对图像目标进行特征提取，得到多帧目标识别数据，并计算检测模块的损失；反向传播进行检测模块的YOLO参数更新；将检测模块得到的多帧目标识别数据输入到关联模块，以检测目标为节点构建图结构，提取高级目标特征，进行下一时刻目标位置的预测，结果计算总的损失；通过梯度反向传播进行关联模块参数的更新，得到训练好的海上目标追踪模型；将测试集输入训练好的海上目标追踪模型，输出检测追踪结果和概率。本发明考虑到不同目标之间的空间关系，目标追踪性能好。