Systems and methods for simulating traffic scenes
Example aspects of the present disclosure describe a scene generator for simulating scenes in an environment. For example, snapshots of simulated traffic scenes can be generated by sampling a joint probability distribution trained on real-world traffic scenes. In some implementations, samples of the joint probability distribution can be obtained by sampling a plurality of factorized probability distributions for a plurality of objects for sequential insertion into the scene.
1 . A computer-implemented method for traffic scene generation, comprising:
(a) obtaining environmental data descriptive of an environment and a subject vehicle within the environment;
(b) generating a first parameter of a new object for insertion into a synthesized traffic scene in the environment, the first parameter generated using a first parameter prediction machine-learned model of a machine-learned traffic scene generation framework and based at least in part on the environmental data;
(c) generating a second parameter of the new object, the second parameter generated using a second parameter prediction machine-learned model of the machine-learned traffic scene generation framework and based at least in part on the environmental data and the first parameter; and
(d) outputting data descriptive of the synthesized traffic scene.
2 . The computer-implemented method of claim 1 , further comprising:
generating feature data using a machine-learned feature extraction model based on a current state of the synthesized traffic scene.
3 . The computer-implemented method of claim 2 , wherein the machine-learned feature extraction model processes multiple prior states of the synthesized traffic scene.
4 . The computer-implemented method of claim 2 , wherein the first parameter and the feature data are provided as inputs to the second parameter prediction machine-learned model.
5 . The computer-implemented method of claim 4 , further comprising:
(e) inputting the first parameter, the second parameter, and the feature data to a third parameter prediction machine-learned model of the machine-learned traffic scene generation framework.
6 . The computer-implemented method of claim 1 , wherein at least one of the first parameter or the second parameter describes at least one of: an object class, an object position, an object orientation, an object bounding box, or an object velocity.
7 . The computer-implemented method of claim 1 , further comprising:
(f) generating simulated sensor data for the environment based on the data descriptive of the synthesized traffic scene;
(g) obtaining labels for the simulated sensor data based on the first parameter and the second parameter; and
(h) training a machine-learned model of an autonomous vehicle control system using the labels and the simulated sensor data.
8 . A computing system for traffic scene generation, the computing system comprising:
one or more processors; and
one or more non-transitory computer-readable media that store instructions for execution by the one or more processors to cause the computing system to perform operations, the operations comprising:
(a) obtaining environmental data descriptive of an environment and a subject vehicle within the environment;
(b) generating a first parameter of a new object for insertion into a synthesized traffic scene in the environment, the first parameter generated using a first parameter prediction machine-learned model of a machine-learned traffic scene generation framework and based at least in part on the environmental data;
(c) generating a second parameter of the new object, the second parameter generated using a second parameter prediction machine-learned model of the machine-learned traffic scene generation framework and based at least in part on the environmental data and the first parameter; and
(d) outputting data descriptive of the synthesized traffic scene.
9 . The computing system of claim 8 , further comprising:
generating feature data using a machine-learned feature extraction model based on a current state of the synthesized traffic scene.
10 . The computing system of claim 9 , wherein the machine-learned feature extraction model processes multiple prior states of the synthesized traffic scene.
11 . The computing system of claim 9 , wherein the first parameter and the feature data are provided as inputs to the second parameter prediction machine-learned model.
12 . The computing system of claim 11 , further comprising:
(e) inputting the first parameter, the second parameter, and the feature data to a third parameter prediction machine-learned model of the machine-learned traffic scene generation framework.
13 . The computing system of claim 8 , wherein at least one of the first parameter or the second parameter describes at least one of: an object class, an object position, an object orientation, an object bounding box, or an object velocity.
14 . The computing system of claim 8 , wherein the operations comprise:
(f) generating simulated sensor data for the environment based on the data descriptive of the synthesized traffic scene;
(g) obtaining labels for the simulated sensor data based on the first parameter and the second parameter; and
(h) training a machine-learned model of an autonomous vehicle control system using the labels and the simulated sensor data.
15 . An autonomous vehicle control system comprising:
one or more machine-learned models that have been trained using simulated sensor data representing at least a portion of a synthesized traffic scene, the simulated sensor data having been generated by performance of operations, the operations comprising:
(a) obtaining environmental data descriptive of an environment and a subject vehicle within the environment;
(b) generating a first parameter of a new object for insertion into a synthesized traffic scene in the environment, the first parameter generated using a first parameter prediction machine-learned model of a machine-learned traffic scene generation framework and based at least in part on the environmental data;
(c) generating a second parameter of the new object, the second parameter generated using a second parameter prediction machine-learned model of the machine-learned traffic scene generation framework and based at least in part on the environmental data and the first parameter; and
(d) outputting data descriptive of the synthesized traffic scene.
16 . The autonomous vehicle control system of claim 15 , the operations further comprising:
generating feature data using a machine-learned feature extraction model based on a current state of the synthesized traffic scene.
17 . The autonomous vehicle control system of claim 16 , wherein the machine-learned feature extraction model processes multiple prior states of the synthesized traffic scene.
18 . The autonomous vehicle control system of claim 16 , wherein the first parameter and the feature data are provided as inputs to the second parameter prediction machine-learned model.
19 . The autonomous vehicle control system of claim 18 , the operations further comprising:
(e) inputting the first parameter, the second parameter, and the feature data to a third parameter prediction machine-learned model of the machine-learned traffic scene generation framework.
20 . The autonomous vehicle control system of claim 15 , the operations further comprising:
(f) generating simulated sensor data for the environment based on the data descriptive of the synthesized traffic scene;
(g) obtaining labels for the simulated sensor data based on the first parameter and the second parameter; and
(h) training at least one of the one or more machine-learned models of the autonomous vehicle control system using the labels and the simulated sensor data.