Index _ | A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | R | S | T | U | V | W | X | Y | Z _ __getitem__() (torchdrivesim.mesh.BaseMesh method) (torchdrivesim.simulator.Simulator method) __post_init__() (torchdrivesim.map.Stopline method) (torchdrivesim.mesh.BaseMesh method) __version__ (in module torchdrivesim) _advance_state() (torchdrivesim.goals.WaypointGoal method) _agent_types (torchdrivesim.simulator.Simulator attribute) _agent_waypoint_overlap() (torchdrivesim.goals.WaypointGoal method) _attrs_fill (torchdrivesim.mesh.AttributeMesh attribute) _batch_size (torchdrivesim.simulator.Simulator attribute) _cat_fill (torchdrivesim.mesh.BirdviewMesh attribute) _compute_collision_of_multi_agents() (torchdrivesim.simulator.Simulator method) _compute_collision_of_single_agent() (torchdrivesim.simulator.Simulator method) _create_traffic_controls_mesh() (torchdrivesim.mesh.BirdviewRGBMeshGenerator class method) _default_allowed_states() (torchdrivesim.traffic_controls.BaseTrafficControl class method) (torchdrivesim.traffic_controls.TrafficLightControl class method) _default_mask() (torchdrivesim.goals.WaypointGoal method) (torchdrivesim.traffic_controls.BaseTrafficControl method) _default_replay_states() (torchdrivesim.traffic_controls.BaseTrafficControl method) _default_state() (torchdrivesim.goals.WaypointGoal method) (torchdrivesim.traffic_controls.BaseTrafficControl method) _deserialize_tensors() (torchdrivesim.mesh.AttributeMesh class method) (torchdrivesim.mesh.BaseMesh class method) (torchdrivesim.mesh.BirdviewMesh class method) _faces_fill (torchdrivesim.mesh.BaseMesh attribute) _filename_defaults() (in module torchdrivesim.map) _make_actors_mesh() (torchdrivesim.mesh.BirdviewRGBMeshGenerator class method) _make_direction_mesh() (torchdrivesim.mesh.BirdviewRGBMeshGenerator class method) _make_waypoint_mesh() (torchdrivesim.mesh.BirdviewRGBMeshGenerator class method) _normalization_factor (torchdrivesim.kinematic.KinematicBicycle attribute) (torchdrivesim.kinematic.SimpleKinematicModel attribute) _resource_path (in module torchdrivesim) _states (torchdrivesim.traffic_lights.TrafficLightStateMachine attribute) _traffic_light_controller (torchdrivesim.behavior.iai.IAINPCController attribute) _trim_and_return_verts_and_faces() (torchdrivesim.mesh.BaseMesh method) _type_ (torchdrivesim.observation_noise.MapObservationNoiseFromLogConfig attribute) (torchdrivesim.observation_noise.ObservationNoiseConfig attribute) (torchdrivesim.observation_noise.StandardSensingObservationNoiseConfig attribute) _update_mask() (torchdrivesim.goals.WaypointGoal method) _validate_input() (torchdrivesim.mesh.AttributeMesh method) (torchdrivesim.mesh.BaseMesh method) (torchdrivesim.mesh.BirdviewMesh method) (torchdrivesim.mesh.RGBMesh method) _verts_fill (torchdrivesim.mesh.BaseMesh attribute) _xy_normalization_tensor (torchdrivesim.kinematic.BicycleByDisplacement attribute) A action_model_extras (torchdrivesim.simulator.Simulator attribute) action_size (torchdrivesim.kinematic.CompoundKinematicModel attribute) (torchdrivesim.kinematic.KinematicBicycle attribute) (torchdrivesim.kinematic.KinematicModel attribute) (torchdrivesim.simulator.Simulator property) actor_id (torchdrivesim.map.Stopline attribute) actor_mesh (torchdrivesim.mesh.BirdviewRGBMeshGenerator attribute) actor_states (torchdrivesim.traffic_lights.TrafficLightGroupState attribute) ActorStates (in module torchdrivesim.traffic_lights) add_static_meshes() (torchdrivesim.mesh.BirdviewRGBMeshGenerator method) add_static_rgb_meshes() (torchdrivesim.mesh.BirdviewRGBMeshGenerator method) advance_npcs() (torchdrivesim.behavior.iai.IAINPCController method) (torchdrivesim.behavior.replay.ReplayController method) (torchdrivesim.simulator.CompoundNPCController method) (torchdrivesim.simulator.NPCController method) agent_attributes_to_basic_agent_properties() (in module torchdrivesim.behavior.iai) agent_count (torchdrivesim.simulator.Simulator property) agent_lr (torchdrivesim.simulator.Simulator attribute) agent_properties_to_agent_attributes() (in module torchdrivesim.behavior.iai) agent_size (torchdrivesim.simulator.Simulator attribute) agent_type (torchdrivesim.map.Stopline attribute) (torchdrivesim.simulator.Simulator attribute) agent_type_names (torchdrivesim.behavior.iai.IAINPCController attribute) (torchdrivesim.simulator.NPCController attribute) agent_types (torchdrivesim.simulator.Simulator property) allowed_states (torchdrivesim.traffic_controls.BaseTrafficControl attribute) antialias (torchdrivesim.rendering.nvdiffrast.NvdiffrastRendererConfig attribute) (torchdrivesim.rendering.NvdiffrastRendererConfig attribute) assert_equal() (in module torchdrivesim.utils) assert_pytorch3d_available() (in module torchdrivesim) attr_dim (torchdrivesim.mesh.AttributeMesh property) AttributeMesh (class in torchdrivesim.mesh) attrs (torchdrivesim.mesh.AttributeMesh attribute) B backend (torchdrivesim.rendering.base.DummyRendererConfig attribute) (torchdrivesim.rendering.base.RendererConfig attribute) (torchdrivesim.rendering.cv2.CV2RendererConfig attribute) (torchdrivesim.rendering.CV2RendererConfig attribute) (torchdrivesim.rendering.DummyRendererConfig attribute) (torchdrivesim.rendering.nvdiffrast.NvdiffrastRendererConfig attribute) (torchdrivesim.rendering.NvdiffrastRendererConfig attribute) (torchdrivesim.rendering.pytorch3d.Pytorch3DRendererConfig attribute) (torchdrivesim.rendering.Pytorch3DRendererConfig attribute) (torchdrivesim.rendering.RendererConfig attribute) background_color (torchdrivesim.rendering.pytorch3d.Shader2D attribute) BadMeshFormat BaseMesh (class in torchdrivesim.mesh) BaseTrafficControl (class in torchdrivesim.traffic_controls) batch_assignments (torchdrivesim.kinematic.CompoundKinematicModel property) batch_shape (torchdrivesim.kinematic.CompoundKinematicModel property) batch_size (torchdrivesim.kinematic.CompoundKinematicModel property) (torchdrivesim.kinematic.KinematicModel property) (torchdrivesim.mesh.BaseMesh property) (torchdrivesim.simulator.Simulator property) bbox2discs() (in module torchdrivesim.infractions) BicycleByDisplacement (class in torchdrivesim.kinematic) BicycleByOrientedDisplacement (class in torchdrivesim.kinematic) BicycleNoReversing (class in torchdrivesim.kinematic) BirdviewMesh (class in torchdrivesim.mesh) BirdviewRenderer (class in torchdrivesim.rendering) (class in torchdrivesim.rendering.base) BirdviewRGBMeshGenerator (class in torchdrivesim.mesh) blend (torchdrivesim.rendering.pytorch3d.Shader2D attribute) box1_in_box2() (in module torchdrivesim._iou_utils) box2corners_th() (in module torchdrivesim._iou_utils) box2corners_with_rear_factor() (in module torchdrivesim._iou_utils) box_in_box_th() (in module torchdrivesim._iou_utils) box_intersection_th() (in module torchdrivesim._iou_utils) build_vertices() (in module torchdrivesim._iou_utils) build_verts_faces_from_bounding_box() (in module torchdrivesim.mesh) C calculate_area() (in module torchdrivesim._iou_utils) Cameras (class in torchdrivesim.rendering.base) categories (torchdrivesim.mesh.BirdviewMesh attribute) center (torchdrivesim.map.MapConfig attribute) (torchdrivesim.mesh.BaseMesh property) cfg (torchdrivesim.observation_noise.ObservationNoise attribute) (torchdrivesim.rendering.base.BirdviewRenderer attribute) (torchdrivesim.rendering.BirdviewRenderer attribute) (torchdrivesim.rendering.cv2.CV2Renderer attribute) (torchdrivesim.rendering.CV2Renderer attribute) (torchdrivesim.rendering.nvdiffrast.NvdiffrastRenderer attribute) (torchdrivesim.rendering.NvdiffrastRenderer attribute) (torchdrivesim.rendering.pytorch3d.Pytorch3DRenderer attribute) (torchdrivesim.rendering.Pytorch3DRenderer attribute) (torchdrivesim.simulator.Simulator attribute) clone() (torchdrivesim.mesh.BaseMesh method) collate() (torchdrivesim.mesh.AttributeMesh class method) (torchdrivesim.mesh.BaseMesh class method) (torchdrivesim.mesh.BirdviewMesh class method) collect_all_current_light_states() (torchdrivesim.traffic_lights.TrafficLightController method) collision_detection_with_discs() (in module torchdrivesim.infractions) collision_metric (torchdrivesim.simulator.TorchDriveConfig attribute) CollisionMetric (class in torchdrivesim.simulator) Color (in module torchdrivesim.mesh) color_map (torchdrivesim.mesh.BirdviewRGBMeshGenerator attribute) (torchdrivesim.rendering.base.BirdviewRenderer attribute) (torchdrivesim.rendering.BirdviewRenderer attribute) colors (torchdrivesim.mesh.BirdviewMesh attribute) CompoundKinematicModel (class in torchdrivesim.kinematic) CompoundNPCController (class in torchdrivesim.simulator) compute_agent_collisions_metric() (in module torchdrivesim.infractions) compute_agent_collisions_metric_pytorch3d() (in module torchdrivesim.infractions) compute_collision() (torchdrivesim.simulator.Simulator method) compute_offroad() (torchdrivesim.simulator.Simulator method) compute_state() (torchdrivesim.traffic_controls.BaseTrafficControl method) compute_traffic_lights_violations() (torchdrivesim.simulator.Simulator method) compute_violation() (torchdrivesim.traffic_controls.BaseTrafficControl method) (torchdrivesim.traffic_controls.TrafficLightControl method) compute_wrong_way() (torchdrivesim.simulator.Simulator method) concat() (torchdrivesim.mesh.AttributeMesh class method) (torchdrivesim.mesh.BaseMesh class method) (torchdrivesim.mesh.BirdviewMesh class method) construct_cameras() (torchdrivesim.rendering.base.BirdviewRenderer method) (torchdrivesim.rendering.BirdviewRenderer method) construct_pytorch3d_cameras() (in module torchdrivesim.rendering.pytorch3d) controller_indices (torchdrivesim.simulator.CompoundNPCController attribute) controllers (torchdrivesim.simulator.CompoundNPCController attribute) copy() (torchdrivesim.behavior.iai.IAINPCController method) (torchdrivesim.behavior.replay.ReplayController method) (torchdrivesim.goals.WaypointGoal method) (torchdrivesim.kinematic.BicycleByDisplacement method) (torchdrivesim.kinematic.CompoundKinematicModel method) (torchdrivesim.kinematic.KinematicBicycle method) (torchdrivesim.kinematic.KinematicModel method) (torchdrivesim.kinematic.SimpleKinematicModel method) (torchdrivesim.lanelet2.LaneFeatures method) (torchdrivesim.mesh.BirdviewRGBMeshGenerator method) (torchdrivesim.rendering.base.BirdviewRenderer method) (torchdrivesim.rendering.BirdviewRenderer method) (torchdrivesim.simulator.CompoundNPCController method) (torchdrivesim.simulator.NPCController method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SpawnController method) (torchdrivesim.traffic_controls.BaseTrafficControl method) corners (torchdrivesim.traffic_controls.BaseTrafficControl attribute) current_light_state_tensor_from_controller() (in module torchdrivesim.traffic_lights) current_state (torchdrivesim.traffic_lights.TrafficLightController property) (torchdrivesim.traffic_lights.TrafficLightStateMachine property) current_state_with_name (torchdrivesim.traffic_lights.TrafficLightController property) CV2Renderer (class in torchdrivesim.rendering) (class in torchdrivesim.rendering.cv2) CV2RendererConfig (class in torchdrivesim.rendering) (class in torchdrivesim.rendering.cv2) D denormalize_action() (torchdrivesim.kinematic.CompoundKinematicModel method) (torchdrivesim.kinematic.KinematicBicycle method) (torchdrivesim.kinematic.KinematicModel method) (torchdrivesim.kinematic.SimpleKinematicModel method) dense_lane_features (torchdrivesim.lanelet2.LaneFeatures attribute) dense_lane_features_mask (torchdrivesim.lanelet2.LaneFeatures attribute) deserialize() (torchdrivesim.mesh.BaseMesh class method) device (torchdrivesim.mesh.BaseMesh property) (torchdrivesim.rendering.base.RendererConfig attribute) (torchdrivesim.rendering.RendererConfig attribute) differentiable_rendering (torchdrivesim.rendering.pytorch3d.Pytorch3DRendererConfig attribute) (torchdrivesim.rendering.Pytorch3DRendererConfig attribute) dim (torchdrivesim.mesh.BaseMesh property) discs (torchdrivesim.simulator.CollisionMetric attribute) download_iai_map() (in module torchdrivesim.map) drive_model_version (torchdrivesim.behavior.iai.IAINPCController attribute) dt (torchdrivesim.kinematic.KinematicModel attribute) DummyRenderer (class in torchdrivesim.rendering) (class in torchdrivesim.rendering.base) DummyRendererConfig (class in torchdrivesim.rendering) (class in torchdrivesim.rendering.base) duration (torchdrivesim.traffic_lights.TrafficLightGroupState attribute) (torchdrivesim.traffic_lights.TrafficLightStateMachine property) E empty() (torchdrivesim.mesh.AttributeMesh class method) (torchdrivesim.mesh.BaseMesh class method) (torchdrivesim.mesh.BirdviewMesh class method) EPSILON (in module torchdrivesim._iou_utils) exit_boundary (torchdrivesim.simulator.SpawnController attribute) expand() (torchdrivesim.mesh.AttributeMesh method) (torchdrivesim.mesh.BaseMesh method) (torchdrivesim.mesh.BirdviewMesh method) (torchdrivesim.mesh.BirdviewRGBMeshGenerator method) extend() (torchdrivesim.behavior.replay.ReplayController method) (torchdrivesim.goals.WaypointGoal method) (torchdrivesim.kinematic.CompoundKinematicModel method) (torchdrivesim.kinematic.KinematicModel method) (torchdrivesim.lanelet2.LaneFeatures method) (torchdrivesim.simulator.CompoundNPCController method) (torchdrivesim.simulator.NPCController method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SpawnController method) (torchdrivesim.traffic_controls.BaseTrafficControl method) F faces (torchdrivesim.mesh.BaseMesh attribute) faces_count (torchdrivesim.mesh.BaseMesh property) fill_attr() (torchdrivesim.mesh.BirdviewMesh method) find_direction() (in module torchdrivesim.lanelet2) find_lanelet_directions() (in module torchdrivesim.lanelet2) find_map_config() (in module torchdrivesim.map) find_wrong_way_stoplines() (in module torchdrivesim.map) fit_action() (torchdrivesim.kinematic.BicycleByDisplacement method) (torchdrivesim.kinematic.BicycleByOrientedDisplacement method) (torchdrivesim.kinematic.CompoundKinematicModel method) (torchdrivesim.kinematic.KinematicBicycle method) (torchdrivesim.kinematic.KinematicModel method) (torchdrivesim.kinematic.OrientedKinematicModel method) (torchdrivesim.kinematic.SimpleKinematicModel method) (torchdrivesim.kinematic.TeleportingKinematicModel method) (torchdrivesim.simulator.Simulator method) flattening() (torchdrivesim.kinematic.CompoundKinematicModel method) (torchdrivesim.kinematic.KinematicBicycle method) (torchdrivesim.kinematic.KinematicModel method) forward() (torchdrivesim.rendering.pytorch3d.Shader2D method) from_json() (torchdrivesim.traffic_lights.TrafficLightController class method) (torchdrivesim.traffic_lights.TrafficLightStateMachine class method) G gather_npc_states() (torchdrivesim.simulator.CompoundNPCController method) generate() (torchdrivesim.mesh.BirdviewRGBMeshGenerator method) generate_annulus_polygon_mesh() (in module torchdrivesim.mesh) generate_disc_mesh() (in module torchdrivesim.mesh) generate_trajectory_mesh() (in module torchdrivesim.mesh) get_action_model_extras() (torchdrivesim.simulator.Simulator method) get_agent_lr() (torchdrivesim.simulator.Simulator method) get_agent_size() (torchdrivesim.simulator.Simulator method) get_agent_type() (torchdrivesim.simulator.Simulator method) get_agent_type_names() (torchdrivesim.simulator.Simulator method) get_all_agent_present_mask() (torchdrivesim.simulator.Simulator method) get_all_agent_size() (torchdrivesim.simulator.Simulator method) get_all_agent_state() (torchdrivesim.simulator.Simulator method) get_all_agent_type() (torchdrivesim.simulator.Simulator method) get_all_agents_absolute() (torchdrivesim.simulator.Simulator method) get_all_agents_relative() (torchdrivesim.simulator.Simulator method) get_all_intersections() (in module torchdrivesim.infractions) get_camera_center() (torchdrivesim.rendering.base.Cameras method) get_color() (torchdrivesim.rendering.base.BirdviewRenderer method) (torchdrivesim.rendering.BirdviewRenderer method) get_current_actor_states() (torchdrivesim.traffic_lights.TrafficLightStateMachine method) get_default_color_map() (in module torchdrivesim.rendering.base) get_default_rendering_levels() (in module torchdrivesim.rendering.base) get_glctx_session() (in module torchdrivesim.rendering.nvdiffrast) get_masks() (torchdrivesim.goals.WaypointGoal method) get_noisy_agent_size() (torchdrivesim.observation_noise.ObservationNoise method) (torchdrivesim.simulator.Simulator method) get_noisy_all_agents_absolute() (torchdrivesim.simulator.Simulator method) get_noisy_all_agents_relative() (torchdrivesim.simulator.Simulator method) get_noisy_background_mesh() (torchdrivesim.observation_noise.MapObservationNoiseFromLog method) (torchdrivesim.observation_noise.ObservationNoise method) (torchdrivesim.simulator.Simulator method) get_noisy_crosswalk_features() (torchdrivesim.observation_noise.MapObservationNoiseFromLog method) get_noisy_lane_features() (torchdrivesim.observation_noise.MapObservationNoiseFromLog method) (torchdrivesim.observation_noise.ObservationNoise method) (torchdrivesim.simulator.Simulator method) get_noisy_present_mask() (torchdrivesim.observation_noise.ObservationNoise method) (torchdrivesim.observation_noise.StandardSensingObservationNoise method) (torchdrivesim.simulator.Simulator method) get_noisy_road_mesh() (torchdrivesim.observation_noise.MapObservationNoiseFromLog method) (torchdrivesim.observation_noise.ObservationNoise method) (torchdrivesim.simulator.Simulator method) get_noisy_state() (torchdrivesim.observation_noise.ObservationNoise method) (torchdrivesim.observation_noise.StandardSensingObservationNoise method) (torchdrivesim.simulator.Simulator method) get_noisy_traffic_controls() (torchdrivesim.observation_noise.MapObservationNoiseFromLog method) (torchdrivesim.observation_noise.ObservationNoise method) (torchdrivesim.simulator.Simulator method) get_npc_present_mask() (torchdrivesim.simulator.NPCController method) (torchdrivesim.simulator.Simulator method) get_npc_size() (torchdrivesim.simulator.NPCController method) (torchdrivesim.simulator.Simulator method) get_npc_state() (torchdrivesim.simulator.NPCController method) (torchdrivesim.simulator.Simulator method) get_npc_types() (torchdrivesim.simulator.NPCController method) (torchdrivesim.simulator.Simulator method) get_number_of_light_groups() (torchdrivesim.traffic_lights.TrafficLightController method) get_params() (torchdrivesim.kinematic.CompoundKinematicModel method) (torchdrivesim.kinematic.KinematicBicycle method) (torchdrivesim.kinematic.KinematicModel method) get_present_mask() (torchdrivesim.simulator.Simulator method) get_state() (torchdrivesim.kinematic.CompoundKinematicModel method) (torchdrivesim.kinematic.KinematicModel method) (torchdrivesim.simulator.Simulator method) get_traffic_controls() (torchdrivesim.simulator.Simulator method) get_view_to_proj_transform() (torchdrivesim.rendering.base.Cameras method) get_waypoints() (torchdrivesim.goals.WaypointGoal method) (torchdrivesim.simulator.Simulator method) get_waypoints_mask() (torchdrivesim.simulator.Simulator method) get_waypoints_state() (torchdrivesim.simulator.Simulator method) get_world_center() (torchdrivesim.simulator.Simulator method) get_world_to_view_transform() (torchdrivesim.rendering.base.Cameras method) glctx (torchdrivesim.rendering.nvdiffrast.NvdiffrastRenderer attribute) (torchdrivesim.rendering.NvdiffrastRenderer attribute) glctx_sessions (in module torchdrivesim.rendering.nvdiffrast) green (torchdrivesim.traffic_lights.TrafficLightState attribute) H hard (torchdrivesim.rendering.pytorch3d.RenderingBlend attribute) height (torchdrivesim.utils.Resolution attribute) heuristic_initialize() (in module torchdrivesim.behavior.heuristic) highlight_ego_vehicle (torchdrivesim.rendering.base.RendererConfig attribute) (torchdrivesim.rendering.RendererConfig attribute) I iai_drive() (in module torchdrivesim.behavior.iai) iai_initialize() (in module torchdrivesim.behavior.iai) iai_location_name (torchdrivesim.map.MapConfig attribute) IAINPCController (class in torchdrivesim.behavior.iai) InitializationFailedError initialize_actors_mesh() (torchdrivesim.mesh.BirdviewRGBMeshGenerator method) initialize_background_mesh() (torchdrivesim.mesh.BirdviewRGBMeshGenerator method) initialize_traffic_controls_mesh() (torchdrivesim.mesh.BirdviewRGBMeshGenerator method) initialize_waypoint_mesh() (torchdrivesim.mesh.BirdviewRGBMeshGenerator method) interaction_replay() (in module torchdrivesim.behavior.replay) internal_time (torchdrivesim.simulator.Simulator attribute) iou (torchdrivesim.simulator.CollisionMetric attribute) iou_differentiable() (in module torchdrivesim.infractions) iou_differentiable_fast() (in module torchdrivesim._iou_utils) iou_non_differentiable() (in module torchdrivesim._iou_utils) is_available (in module torchdrivesim.lanelet2) (in module torchdrivesim.rendering.nvdiffrast) (in module torchdrivesim.rendering.pytorch3d) is_inside_polygon() (in module torchdrivesim.utils) isin() (in module torchdrivesim.utils) K kinematic_model (torchdrivesim.simulator.Simulator attribute) KinematicBicycle (class in torchdrivesim.kinematic) KinematicModel (class in torchdrivesim.kinematic) L lane_features (torchdrivesim.simulator.Simulator attribute) LaneFeatures (class in torchdrivesim.lanelet2) Lanelet2NotFound lanelet_inclusion_tolerance (torchdrivesim.simulator.TorchDriveConfig attribute) lanelet_map (torchdrivesim.map.MapConfig property) (torchdrivesim.simulator.Simulator attribute) lanelet_map_origin (torchdrivesim.map.MapConfig attribute) lanelet_map_to_lane_mesh() (in module torchdrivesim.lanelet2) lanelet_orientation_loss() (in module torchdrivesim.infractions) lanelet_path (torchdrivesim.map.MapConfig attribute) LANELET_TAGS_TO_EXCLUDE (in module torchdrivesim.infractions) LaneletError left_handed (torchdrivesim.kinematic.KinematicBicycle attribute) left_handed_coordinates (torchdrivesim.map.MapConfig attribute) (torchdrivesim.rendering.base.RendererConfig attribute) (torchdrivesim.rendering.RendererConfig attribute) (torchdrivesim.simulator.TorchDriveConfig attribute) length (torchdrivesim.map.Stopline attribute) light_states_all_timesteps (torchdrivesim.behavior.iai.IAINPCController attribute) line_circle_intersection() (in module torchdrivesim.utils) line_segments_to_mesh() (in module torchdrivesim.lanelet2) load() (torchdrivesim.mesh.AttributeMesh class method) (torchdrivesim.mesh.BaseMesh class method) load_lanelet_map() (in module torchdrivesim.lanelet2) load_map_config() (in module torchdrivesim.map) location (torchdrivesim.behavior.iai.IAINPCController attribute) logger (in module torchdrivesim.infractions) (in module torchdrivesim.kinematic) (in module torchdrivesim.mesh) (in module torchdrivesim.rendering) (in module torchdrivesim.rendering.base) (in module torchdrivesim.rendering.nvdiffrast) (in module torchdrivesim.simulator) (in module torchdrivesim.traffic_lights) lr (torchdrivesim.kinematic.KinematicBicycle attribute) M make_renderer() (torchdrivesim.rendering.pytorch3d.Pytorch3DRenderer class method) (torchdrivesim.rendering.Pytorch3DRenderer class method) map_param() (torchdrivesim.kinematic.CompoundKinematicModel method) (torchdrivesim.kinematic.KinematicBicycle method) (torchdrivesim.kinematic.KinematicModel method) MapConfig (class in torchdrivesim.map) MapObservationNoiseFromLog (class in torchdrivesim.observation_noise) MapObservationNoiseFromLogConfig (class in torchdrivesim.observation_noise) mask (torchdrivesim.goals.WaypointGoal attribute) (torchdrivesim.traffic_controls.BaseTrafficControl attribute) max_acceleration (torchdrivesim.kinematic.KinematicBicycle attribute) max_dpsi (torchdrivesim.kinematic.SimpleKinematicModel attribute) max_dv (torchdrivesim.kinematic.SimpleKinematicModel attribute) max_dx (torchdrivesim.kinematic.BicycleByDisplacement attribute) (torchdrivesim.kinematic.SimpleKinematicModel attribute) max_goal_idx (torchdrivesim.goals.WaypointGoal attribute) max_minibatch_size (torchdrivesim.rendering.nvdiffrast.NvdiffrastRendererConfig attribute) (torchdrivesim.rendering.NvdiffrastRendererConfig attribute) max_steering (torchdrivesim.kinematic.KinematicBicycle attribute) merge() (torchdrivesim.mesh.BaseMesh method) merge_dicts() (in module torchdrivesim.utils) mesh_path (torchdrivesim.map.MapConfig attribute) model_assignments (torchdrivesim.kinematic.CompoundKinematicModel attribute) models (torchdrivesim.kinematic.CompoundKinematicModel attribute) module torchdrivesim torchdrivesim._iou_utils torchdrivesim.behavior torchdrivesim.behavior.common torchdrivesim.behavior.heuristic torchdrivesim.behavior.iai torchdrivesim.behavior.replay torchdrivesim.goals torchdrivesim.infractions torchdrivesim.kinematic torchdrivesim.lanelet2 torchdrivesim.map torchdrivesim.mesh torchdrivesim.observation_noise torchdrivesim.rendering torchdrivesim.rendering.base torchdrivesim.rendering.cv2 torchdrivesim.rendering.nvdiffrast torchdrivesim.rendering.pytorch3d torchdrivesim.simulator torchdrivesim.traffic_controls torchdrivesim.traffic_lights torchdrivesim.utils N name (torchdrivesim.map.MapConfig attribute) next_state (torchdrivesim.traffic_lights.TrafficLightGroupState attribute) nograd (torchdrivesim.simulator.CollisionMetric attribute) nograd_pytorch3d (torchdrivesim.simulator.CollisionMetric attribute) noisy_background_mesh (torchdrivesim.observation_noise.MapObservationNoiseFromLog attribute) noisy_crosswalk_features (torchdrivesim.observation_noise.MapObservationNoiseFromLog attribute) noisy_lane_features (torchdrivesim.observation_noise.MapObservationNoiseFromLog attribute) noisy_traffic_controls (torchdrivesim.observation_noise.MapObservationNoiseFromLog attribute) none (torchdrivesim.traffic_lights.TrafficLightState attribute) normalize_action() (torchdrivesim.kinematic.CompoundKinematicModel method) (torchdrivesim.kinematic.KinematicBicycle method) (torchdrivesim.kinematic.KinematicModel method) (torchdrivesim.kinematic.SimpleKinematicModel method) normalize_angle() (in module torchdrivesim.utils) note (torchdrivesim.map.MapConfig attribute) npc_attribute (torchdrivesim.behavior.iai.IAINPCController attribute) npc_controller (torchdrivesim.simulator.Simulator attribute) npc_count (torchdrivesim.simulator.Simulator property) npc_present_mask (torchdrivesim.simulator.NPCController attribute) npc_present_masks (torchdrivesim.behavior.replay.ReplayController attribute) npc_size (torchdrivesim.simulator.NPCController attribute) npc_state (torchdrivesim.simulator.NPCController attribute) npc_state_shape (torchdrivesim.behavior.iai.IAINPCController attribute) npc_states (torchdrivesim.behavior.replay.ReplayController attribute) npc_types (torchdrivesim.simulator.NPCController attribute) NPCController (class in torchdrivesim.simulator) num_categories (torchdrivesim.mesh.BirdviewMesh property) NvdiffrastNotFound NvdiffrastRenderer (class in torchdrivesim.rendering) (class in torchdrivesim.rendering.nvdiffrast) NvdiffrastRendererConfig (class in torchdrivesim.rendering) (class in torchdrivesim.rendering.nvdiffrast) O ObservationNoise (class in torchdrivesim.observation_noise) ObservationNoiseConfig (class in torchdrivesim.observation_noise) offroad_infraction_loss() (in module torchdrivesim.infractions) offroad_threshold (torchdrivesim.simulator.TorchDriveConfig attribute) offset() (torchdrivesim.mesh.BaseMesh method) opengl (torchdrivesim.rendering.nvdiffrast.NvdiffrastRendererConfig attribute) (torchdrivesim.rendering.NvdiffrastRendererConfig attribute) orientation (torchdrivesim.map.Stopline attribute) oriented_box_intersection_2d() (in module torchdrivesim._iou_utils) OrientedKinematicModel (class in torchdrivesim.kinematic) P pack_state() (torchdrivesim.kinematic.KinematicModel static method) pad() (torchdrivesim.mesh.AttributeMesh method) (torchdrivesim.mesh.BaseMesh method) (torchdrivesim.mesh.BirdviewMesh method) pick_random_point_and_orientation() (in module torchdrivesim.lanelet2) pickle() (torchdrivesim.mesh.BaseMesh method) point_mesh_face_distance() (in module torchdrivesim.infractions) point_to_mesh_distance_pt() (in module torchdrivesim.infractions) pos (torchdrivesim.traffic_controls.BaseTrafficControl attribute) precision_rounding() (in module torchdrivesim._iou_utils) present_mask (torchdrivesim.simulator.Simulator attribute) project_world_to_clip_space() (torchdrivesim.rendering.base.Cameras method) pytorch3d() (torchdrivesim.mesh.AttributeMesh method) (torchdrivesim.mesh.BaseMesh method) (torchdrivesim.mesh.BirdviewMesh method) Pytorch3DNotFound Pytorch3DRenderer (class in torchdrivesim.rendering) (class in torchdrivesim.rendering.pytorch3d) Pytorch3DRendererConfig (class in torchdrivesim.rendering) (class in torchdrivesim.rendering.pytorch3d) R recenter_offset (torchdrivesim.simulator.Simulator attribute) rectangle_vertices() (in module torchdrivesim.infractions) recurrent_state (torchdrivesim.behavior.iai.IAINPCController attribute) red (torchdrivesim.traffic_lights.TrafficLightState attribute) relative() (in module torchdrivesim.utils) render() (torchdrivesim.simulator.Simulator method) render_agent_direction (torchdrivesim.rendering.base.RendererConfig attribute) (torchdrivesim.rendering.RendererConfig attribute) render_egocentric() (torchdrivesim.simulator.Simulator method) render_frame() (torchdrivesim.rendering.base.BirdviewRenderer method) (torchdrivesim.rendering.BirdviewRenderer method) render_rgb_mesh() (torchdrivesim.rendering.base.BirdviewRenderer method) (torchdrivesim.rendering.base.DummyRenderer method) (torchdrivesim.rendering.BirdviewRenderer method) (torchdrivesim.rendering.cv2.CV2Renderer method) (torchdrivesim.rendering.CV2Renderer method) (torchdrivesim.rendering.DummyRenderer method) (torchdrivesim.rendering.nvdiffrast.NvdiffrastRenderer method) (torchdrivesim.rendering.NvdiffrastRenderer method) (torchdrivesim.rendering.pytorch3d.Pytorch3DRenderer method) (torchdrivesim.rendering.Pytorch3DRenderer method) renderer (torchdrivesim.rendering.pytorch3d.Pytorch3DRenderer attribute) (torchdrivesim.rendering.Pytorch3DRenderer attribute) (torchdrivesim.simulator.TorchDriveConfig attribute) renderer_from_config() (in module torchdrivesim.rendering) RendererConfig (class in torchdrivesim.rendering) (class in torchdrivesim.rendering.base) rendering_levels (torchdrivesim.mesh.BirdviewRGBMeshGenerator attribute) (torchdrivesim.rendering.base.BirdviewRenderer attribute) (torchdrivesim.rendering.BirdviewRenderer attribute) rendering_mesh() (in module torchdrivesim.mesh) RenderingBlend (class in torchdrivesim.rendering.pytorch3d) replay_states (torchdrivesim.traffic_controls.BaseTrafficControl attribute) ReplayController (class in torchdrivesim.behavior.replay) res (torchdrivesim.rendering.base.BirdviewRenderer attribute) (torchdrivesim.rendering.BirdviewRenderer attribute) reset() (torchdrivesim.traffic_lights.TrafficLightController method) (torchdrivesim.traffic_lights.TrafficLightStateMachine method) Resolution (class in torchdrivesim.utils) resolve_paths_to_absolute() (in module torchdrivesim.map) reverse_transform_points_screen() (torchdrivesim.rendering.base.Cameras method) RGBMesh (class in torchdrivesim.mesh) road_mesh (torchdrivesim.map.MapConfig property) (torchdrivesim.simulator.Simulator attribute) road_mesh_from_lanelet_map() (in module torchdrivesim.lanelet2) rotate() (in module torchdrivesim.utils) rotation_matrix() (in module torchdrivesim.utils) S save() (torchdrivesim.mesh.BaseMesh method) sc (torchdrivesim.rendering.base.Cameras attribute) scale (torchdrivesim.rendering.base.BirdviewRenderer attribute) (torchdrivesim.rendering.base.Cameras attribute) (torchdrivesim.rendering.BirdviewRenderer attribute) select_batch_elements() (torchdrivesim.behavior.replay.ReplayController method) (torchdrivesim.goals.WaypointGoal method) (torchdrivesim.kinematic.CompoundKinematicModel method) (torchdrivesim.kinematic.KinematicModel method) (torchdrivesim.lanelet2.LaneFeatures method) (torchdrivesim.mesh.AttributeMesh method) (torchdrivesim.mesh.BaseMesh method) (torchdrivesim.mesh.BirdviewMesh method) (torchdrivesim.mesh.BirdviewRGBMeshGenerator method) (torchdrivesim.simulator.CompoundNPCController method) (torchdrivesim.simulator.NPCController method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SpawnController method) (torchdrivesim.traffic_controls.BaseTrafficControl method) separate_by_category() (torchdrivesim.mesh.BirdviewMesh method) sequence_number (torchdrivesim.traffic_lights.TrafficLightGroupState attribute) serialize() (torchdrivesim.mesh.AttributeMesh method) (torchdrivesim.mesh.BaseMesh method) (torchdrivesim.mesh.BirdviewMesh method) set_attr() (torchdrivesim.mesh.AttributeMesh class method) set_color() (torchdrivesim.mesh.RGBMesh class method) set_colors_with_defaults() (in module torchdrivesim.mesh) set_params() (torchdrivesim.kinematic.CompoundKinematicModel method) (torchdrivesim.kinematic.KinematicBicycle method) (torchdrivesim.kinematic.KinematicModel method) set_properties() (torchdrivesim.mesh.BirdviewMesh class method) set_state() (torchdrivesim.kinematic.CompoundKinematicModel method) (torchdrivesim.kinematic.KinematicModel method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.traffic_controls.BaseTrafficControl method) set_to() (torchdrivesim.traffic_lights.TrafficLightController method) (torchdrivesim.traffic_lights.TrafficLightStateMachine method) Shader2D (class in torchdrivesim.rendering.pytorch3d) shift_mesh_by_camera_before_rendering (torchdrivesim.rendering.base.RendererConfig attribute) (torchdrivesim.rendering.RendererConfig attribute) sigmoid (torchdrivesim.rendering.pytorch3d.RenderingBlend attribute) SimpleKinematicModel (class in torchdrivesim.kinematic) Simulator (class in torchdrivesim.simulator) single_agent_rendering (torchdrivesim.simulator.TorchDriveConfig attribute) soft (torchdrivesim.rendering.pytorch3d.RenderingBlend attribute) sort_indices() (in module torchdrivesim._iou_utils) sparse_lane_features (torchdrivesim.lanelet2.LaneFeatures attribute) sparse_lane_features_mask (torchdrivesim.lanelet2.LaneFeatures attribute) spawn_controller (torchdrivesim.simulator.NPCController attribute) spawn_despawn_npcs() (torchdrivesim.simulator.NPCController method) (torchdrivesim.simulator.SpawnController method) spawn_masks (torchdrivesim.simulator.SpawnController attribute) spawn_states (torchdrivesim.simulator.SpawnController attribute) SpawnController (class in torchdrivesim.simulator) StandardSensingObservationNoise (class in torchdrivesim.observation_noise) StandardSensingObservationNoiseConfig (class in torchdrivesim.observation_noise) state (torchdrivesim.goals.WaypointGoal attribute) (torchdrivesim.kinematic.KinematicModel attribute) (torchdrivesim.traffic_controls.BaseTrafficControl attribute) state_per_machine (torchdrivesim.traffic_lights.TrafficLightController property) state_size (torchdrivesim.kinematic.CompoundKinematicModel attribute) (torchdrivesim.kinematic.KinematicModel attribute) states (torchdrivesim.traffic_lights.TrafficLightStateMachine property) static_traffic_controls_mesh (torchdrivesim.mesh.BirdviewRGBMeshGenerator attribute) step() (torchdrivesim.goals.WaypointGoal method) (torchdrivesim.kinematic.BicycleByDisplacement method) (torchdrivesim.kinematic.BicycleNoReversing method) (torchdrivesim.kinematic.CompoundKinematicModel method) (torchdrivesim.kinematic.KinematicBicycle method) (torchdrivesim.kinematic.KinematicModel method) (torchdrivesim.kinematic.OrientedKinematicModel method) (torchdrivesim.kinematic.SimpleKinematicModel method) (torchdrivesim.kinematic.TeleportingKinematicModel method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.traffic_controls.BaseTrafficControl method) step_from_xy() (torchdrivesim.kinematic.BicycleByDisplacement method) (torchdrivesim.kinematic.BicycleByOrientedDisplacement method) Stopline (class in torchdrivesim.map) stoplines (torchdrivesim.map.MapConfig property) stoplines_path (torchdrivesim.map.MapConfig attribute) StopSignControl (class in torchdrivesim.traffic_controls) store_map_config() (in module torchdrivesim.map) T TeleportingKinematicModel (class in torchdrivesim.kinematic) tensor_color() (in module torchdrivesim.mesh) tick() (torchdrivesim.traffic_lights.TrafficLightController method) (torchdrivesim.traffic_lights.TrafficLightStateMachine method) time (torchdrivesim.behavior.iai.IAINPCController attribute) (torchdrivesim.behavior.replay.ReplayController attribute) (torchdrivesim.simulator.SpawnController attribute) time_remaining (torchdrivesim.traffic_lights.TrafficLightController property) (torchdrivesim.traffic_lights.TrafficLightStateMachine property) to() (torchdrivesim.behavior.iai.IAINPCController method) (torchdrivesim.behavior.replay.ReplayController method) (torchdrivesim.goals.WaypointGoal method) (torchdrivesim.kinematic.BicycleByDisplacement method) (torchdrivesim.kinematic.CompoundKinematicModel method) (torchdrivesim.kinematic.KinematicBicycle method) (torchdrivesim.kinematic.KinematicModel method) (torchdrivesim.kinematic.SimpleKinematicModel method) (torchdrivesim.lanelet2.LaneFeatures method) (torchdrivesim.mesh.AttributeMesh method) (torchdrivesim.mesh.BaseMesh method) (torchdrivesim.mesh.BirdviewMesh method) (torchdrivesim.mesh.BirdviewRGBMeshGenerator method) (torchdrivesim.simulator.CompoundNPCController method) (torchdrivesim.simulator.NPCController method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SpawnController method) (torchdrivesim.traffic_controls.BaseTrafficControl method) to_json() (torchdrivesim.traffic_lights.TrafficLightController method) (torchdrivesim.traffic_lights.TrafficLightStateMachine method) TorchDriveConfig (class in torchdrivesim.simulator) torchdrivesim module torchdrivesim._iou_utils module torchdrivesim.behavior module torchdrivesim.behavior.common module torchdrivesim.behavior.heuristic module torchdrivesim.behavior.iai module torchdrivesim.behavior.replay module torchdrivesim.goals module torchdrivesim.infractions module torchdrivesim.kinematic module torchdrivesim.lanelet2 module torchdrivesim.map module torchdrivesim.mesh module torchdrivesim.observation_noise module torchdrivesim.rendering module torchdrivesim.rendering.base module torchdrivesim.rendering.cv2 module torchdrivesim.rendering.nvdiffrast module torchdrivesim.rendering.pytorch3d module torchdrivesim.simulator module torchdrivesim.traffic_controls module torchdrivesim.traffic_lights module torchdrivesim.utils module total_replay_time (torchdrivesim.traffic_controls.BaseTrafficControl property) traffic_controls (torchdrivesim.simulator.Simulator attribute) traffic_controls_from_map_config() (in module torchdrivesim.map) traffic_fsms (torchdrivesim.traffic_lights.TrafficLightController attribute) traffic_light_colors (torchdrivesim.mesh.BirdviewRGBMeshGenerator attribute) traffic_light_controller (torchdrivesim.map.MapConfig property) traffic_light_controller_path (torchdrivesim.map.MapConfig attribute) traffic_lights_mesh (torchdrivesim.mesh.BirdviewRGBMeshGenerator attribute) TrafficLightControl (class in torchdrivesim.traffic_controls) TrafficLightController (class in torchdrivesim.traffic_lights) TrafficLightGroupState (class in torchdrivesim.traffic_lights) TrafficLightState (class in torchdrivesim.traffic_lights) TrafficLightStateMachine (class in torchdrivesim.traffic_lights) transform() (in module torchdrivesim.utils) transform_points_screen() (torchdrivesim.rendering.base.Cameras method) translate() (torchdrivesim.mesh.BaseMesh method) trim() (torchdrivesim.mesh.AttributeMesh method) (torchdrivesim.mesh.BaseMesh method) (torchdrivesim.mesh.BirdviewMesh method) trim_mesh_before_rendering (torchdrivesim.rendering.cv2.CV2RendererConfig attribute) (torchdrivesim.rendering.CV2RendererConfig attribute) U unflattening() (torchdrivesim.kinematic.CompoundKinematicModel method) (torchdrivesim.kinematic.KinematicBicycle method) (torchdrivesim.kinematic.KinematicModel method) unify() (torchdrivesim.mesh.BirdviewMesh class method) unpack_attributes() (in module torchdrivesim.behavior.iai) unpack_state() (torchdrivesim.kinematic.KinematicModel static method) unpickle() (torchdrivesim.mesh.AttributeMesh class method) (torchdrivesim.mesh.BaseMesh class method) (torchdrivesim.mesh.BirdviewMesh class method) update_current_state_and_time() (torchdrivesim.traffic_lights.TrafficLightController method) update_present_mask() (torchdrivesim.simulator.Simulator method) V validate_agent_types() (torchdrivesim.simulator.Simulator method) validate_tensor_shapes() (torchdrivesim.simulator.Simulator method) vert_category (torchdrivesim.mesh.BirdviewMesh attribute) verts (torchdrivesim.mesh.BaseMesh attribute) verts_count (torchdrivesim.mesh.BaseMesh property) violation_rear_factor (torchdrivesim.traffic_controls.TrafficLightControl attribute) W warned_no_lanelet (torchdrivesim.simulator.Simulator attribute) waypoint_goals (torchdrivesim.simulator.Simulator attribute) waypoint_removal_threshold (torchdrivesim.simulator.TorchDriveConfig attribute) WaypointGoal (class in torchdrivesim.goals) waypoints (torchdrivesim.goals.WaypointGoal attribute) width (torchdrivesim.map.Stopline attribute) (torchdrivesim.utils.Resolution attribute) world_to_clip_transform (torchdrivesim.rendering.base.Cameras attribute) wrong_way_angle_threshold (torchdrivesim.simulator.TorchDriveConfig attribute) X x (torchdrivesim.map.Stopline attribute) xy (torchdrivesim.rendering.base.Cameras attribute) Y y (torchdrivesim.map.Stopline attribute) yellow (torchdrivesim.traffic_lights.TrafficLightState attribute) YieldControl (class in torchdrivesim.traffic_controls) Z zs (torchdrivesim.mesh.BirdviewMesh attribute)