torchdrivesim.behavior.replay

Classes

ReplayController

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

Functions

interaction_replay(location, dataset_path[, ...])

Module Contents

torchdrivesim.behavior.replay.interaction_replay(location, dataset_path, initial_frame=1, segment_length=40, recording=0)

class torchdrivesim.behavior.replay.ReplayController(npc_size, npc_states, npc_present_masks: torch.Tensor | None = None, time: int = 0, npc_types: torch.Tensor | None = None, agent_type_names: List[str] | 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.

time = 0

npc_states

npc_present_masks = None

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

to(device)

copy()

extend(n, in_place=True)

select_batch_elements(idx, in_place=True)