一种海上落水人员漂流轨迹预测方法

本发明属于海上搜救技术领域，具体公开了一种海上落水人员漂流轨迹预测方法，包括以下步骤：S1、收集海上落水人员相关数据；S2、对数据进行归一化处理；S3、根据处理后的数据集制作训练和测试的数据Loader；S4、循环迭代多次，将每个batch中的数据喂入AMWM‑Transformer模型中，获得模型的输出和原始的标签数据进行loss值计算，根据loss值更新模型参数，完成模型训练；S5、根据已训练好的模型，将测试的输入特征喂入得到输出，与真实标签做loss值进行累加计算，得到最终测试数据的总loss值，使用ADE和FDE对实验结果进行评价。