Systems and methods for generating motion forecast data for actors with respect to an autonomous vehicle and training a machine learned model for the same
Systems and methods for generating motion forecast data for actors with respect to an autonomous vehicle and training a machine learned model for the same are disclosed. The computing system can include an object detection model and a graph neural network including a plurality of nodes and a plurality of edges. The computing system can be configured to input sensor data into the object detection model; receive object detection data describing the location of the plurality of the actors relative to the autonomous vehicle as an output of the object detection model; input the object detection data into the graph neural network; iteratively update a plurality of node states respectively associated with the plurality of nodes; and receive, as an output of the graph neural network, the motion forecast data with respect to the plurality of actors.
1. A computer-implemented method comprising:
generating, using an object detection model, object detection data describing a location of an actor within an environment of an autonomous vehicle;
generating, using a graph neural network, motion forecast data associated with the actor, wherein:
the graph neural network comprises a plurality of nodes, wherein a first node of the plurality of nodes corresponds to the actor, and
the graph neural network is configured to model anticipated interactions between the actor and other actors in the environment by passing one or more messages among the plurality of nodes to update at least one node state of the first node, wherein the anticipated interactions comprise an interaction between the actor and at least one other actor which alters a trajectory of the actor or the at least one other actor in the environment;
determining a motion plan for the autonomous vehicle based on the motion forecast data; and
controlling the autonomous vehicle based on the motion plan.
2. The computer-implemented method of claim 1 , further comprising:
inputting at least one of: (i) sensor data or (ii) map data into the object detection model.
3. The computer-implemented method of claim 2 , wherein the sensor data is captured by a remote computing system.
4. The computer-implemented method of claim 2 , wherein the sensor data and the map data are concatenated prior to being input into the object detection model.
5. The computer-implemented method of claim 1 , wherein the object detection data comprises a region of interest within the environment, the region of interest associated with the actor.
6. The computer-implemented method of claim 1 , wherein at least a portion of the plurality of nodes correspond to the other actors in the environment.
7. The computer-implemented method of claim 1 , wherein the anticipated interactions comprise an interaction within a threshold distance between the actor and at least one other actor in the environment.
8. A computing system comprising:
one or more processors, and
one or more memory resources storing instructions executable by the one or more processors to cause the one or more processors to:
generate, using an object detection model, object detection data describing a location of an actor within an environment of an autonomous vehicle;
generate, using a graph neural network, motion forecast data associated with the actor, wherein:
the graph neural network comprises a plurality of nodes, wherein a first node of the plurality of nodes corresponds to the actor, and
the graph neural network is configured to model anticipated interactions between the actor and other actors in the environment by passing one or more messages among the plurality of nodes to update at least one node state of the first node, wherein the anticipated interactions comprise an interaction between the actor and at least one other actor which alters a trajectory of the actor or the at least one other actor in the environment;
determine a motion plan for the autonomous vehicle based on the motion forecast data; and
control the autonomous vehicle based on the motion plan.
9. The computing system of claim 8 , wherein the one or more processors:
input at least one of: (i) sensor data or (ii) map data into the object detection model.
10. The computing system of claim 9 , wherein the sensor data is captured by a remote computing system.
11. The computing system of claim 9 , wherein the sensor data and the map data are concatenated prior to being input into the object detection model.
12. The computing system of claim 8 , wherein the object detection data comprises a region of interest within the environment, the region of interest associated with the actor.
13. The computing system of claim 8 , wherein at least a portion of the plurality of nodes correspond to the other actors in the environment.
14. The computing system of claim 8 , wherein the anticipated interactions comprise an interaction within a threshold distance between the actor and at least one other actor in the environment.
15. A non-transitory computer-readable media storing instructions executable by one or more processors of an autonomous vehicle computing system to cause the one or more processors to:
generate, using an object detection model, object detection data describing a location of an actor within an environment of an autonomous vehicle;
generate, using a graph neural network, motion forecast data associated with the actor, wherein:
the graph neural network comprises a plurality of nodes, wherein a first node of the plurality of nodes corresponds to the actor, and
the graph neural network is configured to model anticipated interactions between the actor and other actors in the environment by passing one or more messages among the plurality of nodes to update at least one node state of the first node, wherein the anticipated interactions comprise an interaction between the actor and at least one other actor which alters a trajectory of the actor or the at least one other actor in the environment;
determine a motion plan for the autonomous vehicle based on the motion forecast data; and
control the autonomous vehicle based on the motion plan.
16. The non-transitory computer-readable media of claim 15 , wherein the one or more processors:
input at least one of: (i) sensor data or (ii) map data into the object detection model.
17. The non-transitory computer-readable media of claim 16 , wherein the sensor data is captured by a remote computing system.
18. The non-transitory computer-readable media of claim 16 , wherein the sensor data and the map data are concatenated prior to being input into the object detection model.
19. The non-transitory computer-readable media of claim 15 , wherein the object detection data comprises a region of interest within the environment, the region of interest associated with the actor.
20. The non-transitory computer-readable media of claim 15 , wherein at least a portion of the plurality of nodes correspond to the other actors in the environment.