torchdrivesim.behavior.iai

Classes

IAINPCController

Base class for non-playable character controllers. It leaves the state unchanged on each step.

Functions

unpack_attributes(→ torch.Tensor)
agent_attributes_to_basic_agent_properties(→ Dict[str, ...)
agent_properties_to_agent_attributes(→ torch.Tensor)
iai_initialize(location, agent_count[, center])
iai_drive(location, agent_states, agent_attributes, ...)

Module Contents

torchdrivesim.behavior.iai.unpack_attributes(attributes) → torch.Tensor

torchdrivesim.behavior.iai.agent_attributes_to_basic_agent_properties(agent_attributes: torch.Tensor) → Dict[str, float]

torchdrivesim.behavior.iai.agent_properties_to_agent_attributes(agent_properties: Dict[str, float]) → torch.Tensor

torchdrivesim.behavior.iai.iai_initialize(location, agent_count, center=(0, 0), traffic_light_state_history: List[Dict[int, invertedai.common.TrafficLightState]] | None = None)

torchdrivesim.behavior.iai.iai_drive(location: str, agent_states: torch.Tensor, agent_attributes: torch.Tensor, recurrent_states: List[invertedai.common.RecurrentState], traffic_lights_states: Dict[int, invertedai.common.TrafficLightState] | None = None)

class torchdrivesim.behavior.iai.IAINPCController(npc_size, npc_state, npc_lr, location, drive_model_version: str | None = None, npc_present_mask: torch.Tensor | None = None, time: int = 0, npc_types: torch.Tensor | None = None, agent_type_names: List[str] | None = None, traffic_light_controller: torchdrivesim.traffic_lights.TrafficLightController | None = None, light_states_all_timesteps: List[Dict[int, invertedai.common.TrafficLightState]] | None = None, spawn_controller: torchdrivesim.simulator.SpawnController | None = None)

Bases: torchdrivesim.simulator.NPCController

Base class for non-playable character controllers. It leaves the state unchanged on each step.

location

agent_type_names = None

npc_attribute

recurrent_state = None

drive_model_version = None

npc_state_shape

_traffic_light_controller = None

light_states_all_timesteps = None

time = 0

advance_npcs(simulator: torchdrivesim.simulator.Simulator) → None

copy()

Create a deep copy of the controller.

to(device)