Target pedestrian trajectory prediction model training method and device, electronic equipment and storage medium

Abstract

The invention discloses a target pedestrian trajectory prediction model training method and device, electronic equipment and a storage medium, and the target pedestrian trajectory prediction model training method comprises the steps: inputting a first historical position vector set and a second historical position vector set into a target pedestrian trajectory prediction model, and obtaining a prediction position vector set; inputting the real position vector set and the prediction position vector set into a discrimination model to obtain a loss value corresponding to the distance set; and ifthe loss value does not meet the preset condition, training parameters in the target pedestrian trajectory prediction model based on the loss value until the loss value meets the preset condition, wherein the first historical position vector set is composed of first historical position vectors of the target pedestrian at a plurality of historical preset moments, and the second historical positionvector set is composed of a second historical position vector sub-set corresponding to each surrounding pedestrian, so that the target pedestrian trajectory prediction model can be trained based on the data features in the time dimension and the space dimension, and the prediction capability of the target pedestrian trajectory prediction model can be improved.

Description

technical field [0001] The present application relates to the field of pedestrian trajectory prediction, in particular to a target pedestrian trajectory prediction model training method, device, electronic equipment and storage medium. Background technique [0002] Pedestrian trajectory prediction refers to predicting the future trajectory of pedestrians based on their historical movement paths. It has a wide range of application scenarios and is widely used in the fields of intelligent driving, mobile travel, and service robot navigation. [0003] At present, in related technologies, the trajectory prediction method of pedestrians can be to input the historical trajectory of pedestrians into the trained model, and output the predicted trajectory of pedestrians; the process of model training is: input the trajectory of a large number of pedestrians into the model, Train the model. However, many existing pedestrian trajectory prediction methods often use manual design of man...

AI Technical Summary

Problems solved by technology

[0004] The embodiment of the present application aims to solve the technical problem that the trainin

Images