一种基于改进运动模型和Re-ID的无人艇跟踪方法及其设备和可读存储介质

    公开(公告)号:CN118644519A

    公开(公告)日:2024-09-13

    申请号:CN202410490378.8

    申请日:2024-04-23

    Applicant: 上海大学

    Abstract: 本发明属于无人艇跟踪技术领域,具体公开了一种基于改进运动模型和Re‑ID的无人艇跟踪方法及其设备和可读存储介质,其方法包括S1、制作无人艇数据集,获取包括无人艇信息的视频数据;S2、采用YOLOV5模型对步骤S1中得到的视频数据进行目标检测,定位目标在视频数据中的位置;S3、构建基于交互式平方根容积卡尔曼的轨迹预测算法,预测目标在视频数据中下一帧的位置,输出结果;S4、构建基于改进运动模型和Re‑ID目标检测和跟踪方法的重识别模型,提取无人艇的表观特征,对步骤S3中输出的结果进行算法匹配,输出跟踪结果;S5、对模型进行训练;模型训练包括无人艇检测模型以及表观模型的训练,用于训练合适的模型权重。

Patent Agency Ranking