src.tasks.simulate module

src.tasks.simulate module#

class src.tasks.simulate.SimulateState(df_ped: pandas.DataFrame, df_veh: pandas.DataFrame, map_data: object, ped_list: List[int], veh_list: List[int], frame: int, pos_now: torch.Tensor, vel_now: torch.Tensor, hst_now: torch.Tensor, des_now: torch.Tensor, spd_now: torch.Tensor, veh_now: torch.Tensor)[source]#

Bases: object

df_ped: pandas.DataFrame#
df_veh: pandas.DataFrame#
map_data: object#
ped_list: List[int]#
veh_list: List[int]#
frame: int#
pos_now: torch.Tensor#
vel_now: torch.Tensor#
hst_now: torch.Tensor#
des_now: torch.Tensor#
spd_now: torch.Tensor#
veh_now: torch.Tensor#
__init__(df_ped: pandas.DataFrame, df_veh: pandas.DataFrame, map_data: object, ped_list: List[int], veh_list: List[int], frame: int, pos_now: torch.Tensor, vel_now: torch.Tensor, hst_now: torch.Tensor, des_now: torch.Tensor, spd_now: torch.Tensor, veh_now: torch.Tensor) → None#
src.tasks.simulate.init_simulation(args: Namespace, dataset: BaseDataset, frame_idx: int, model: Model) → SimulateState[source]#

Initialize the state required for simulation.

Parameters:

  • args – Configuration arguments.

  • dataset – Dataset object.

  • frame_idx – Initial frame index.

  • model – Model object.

Returns:

Initialized simulation state.

Return type:

SimulateState

src.tasks.simulate.simulate_one_step(args: Namespace, model: Model, diffusion: DDPM, state: SimulateState) → Tuple[pandas.DataFrame, SimulateState][source]#
Contents