Index

_

__init__() (src.dataset.base_dataset.BaseDataset method) (src.dataset.base_dataset.RasterizedMap method) (src.diffusion.ddim.DDIM method) (src.diffusion.ddpm.DDPM method) (src.model.fourier_positional_encoding.FourierPositionalEncoding method) (src.model.mean_pooling_lstm.MeanPoolingLSTM method) (src.model.model.Model method) (src.model.multiscale_cnn.MultiScaleCNN method) (src.model.nan_embedding.NanEmbedding method) (src.model.new_model.NewModel method) (src.model.permuted.Permuted method) (src.model.relative_model.RelativeModel method) (src.model.residual.Residual method) (src.model.sinusoidal_embedding.SinusoidalEmbedding method) (src.tasks.simulate.SimulateState method) (src.utils.auto_gpu.AutoGPU method) (src.utils.plot.EqualizeNormalize method) (src.utils.timer.AbsTimer method) (src.utils.timer.NamedTimer method) (src.utils.timer.ParallelTimer method) (src.utils.timer.Timer method)

A

AbsTimer (class in src.utils.timer) add() (src.utils.timer.AbsTimer method) (src.utils.timer.NamedTimer method) (src.utils.timer.ParallelTimer method) (src.utils.timer.Timer method)

add_minus_flags() (in module src.utils.fix_parser) add_negation_flags() (in module src.utils.fix_parser) add_noise() (src.diffusion.ddpm.DDPM method) affine_transformation() (in module src.utils.homography) allocate_gpu() (src.utils.auto_gpu.AutoGPU static method) AutoGPU (class in src.utils.auto_gpu)

B

BaseDataset (class in src.dataset.base_dataset)

C

calc_homography_mat() (in module src.utils.homography) calc_xy_error() (in module src.utils.calc_xy_error) choice_gpu() (src.utils.auto_gpu.AutoGPU method) clean_trajectory() (src.dataset.sdd_dataset.SDDDataset static method) clear() (src.utils.timer.AbsTimer method) (src.utils.timer.NamedTimer method) (src.utils.timer.ParallelTimer method) (src.utils.timer.Timer method)

clear_svg() (in module src.utils.plot) collate_fn() (src.dataset.base_dataset.BaseDataset static method) cosine_beta_schedule() (src.diffusion.ddpm.DDPM static method) count (src.utils.timer.NamedTimer property) (src.utils.timer.ParallelTimer property) crop_axes_region() (in module src.utils.crop_axes_region)

D

DDIM (class in src.diffusion.ddim) DDPM (class in src.diffusion.ddpm) denoise() (src.diffusion.ddim.DDIM method) (src.diffusion.ddpm.DDPM method)

des_now (src.tasks.simulate.SimulateState attribute) df_ped (src.tasks.simulate.SimulateState attribute) df_veh (src.tasks.simulate.SimulateState attribute)

E

EmptyDatasetError EqualizeNormalize (class in src.utils.plot)

ETHDataset (class in src.dataset.eth_dataset) extract_patches_numpy() (in module src.utils.extract_patches) extract_patches_torch() (in module src.utils.extract_patches)

F

filter_short_trajectories() (src.dataset.waymo_dataset.WayMoDataset static method) filter_vehicle_trajectories() (src.dataset.waymo_dataset.WayMoDataset static method) forward() (src.model.fourier_positional_encoding.FourierPositionalEncoding method) (src.model.mean_pooling_lstm.MeanPoolingLSTM method) (src.model.model.Model method) (src.model.multiscale_cnn.MultiScaleCNN method) (src.model.nan_embedding.NanEmbedding method) (src.model.new_model.NewModel method) (src.model.permuted.Permuted method) (src.model.relative_model.RelativeModel method) (src.model.residual.Residual method) (src.model.sinusoidal_embedding.SinusoidalEmbedding method)

FourierPositionalEncoding (class in src.model.fourier_positional_encoding) frame (src.tasks.simulate.SimulateState attribute)

G

GCDataset (class in src.dataset.gc_dataset) get_abnormal() (src.dataset.gc_dataset.GCDataset class method) get_collision_rate() (in module src.tasks.test_once) get_collision_rate2() (in module src.tasks.test_once) get_fig() (in module src.utils.plot)

get_force_map() (in module src.utils.get_force_map) get_homography_mat() (src.dataset.gc_dataset.GCDataset class method) (src.dataset.sdd_dataset.SDDDataset static method) (src.dataset.ucy_dataset.UCYDataset static method) get_xy_error() (in module src.tasks.test_once) guidance() (in module src.tasks.guidance)

H

hst_now (src.tasks.simulate.SimulateState attribute)

I

image_to_world() (in module src.utils.homography) init_logger() (in module src.utils.logger)

init_simulation() (in module src.tasks.simulate) inverse() (src.utils.plot.EqualizeNormalize method)

J

json_compatible() (in module src.utils.json_compatible)

L

linear_beta_schedule() (src.diffusion.ddpm.DDPM static method) load_cache() (src.dataset.base_dataset.BaseDataset static method) load_data() (src.dataset.base_dataset.BaseDataset class method) (src.dataset.eth_dataset.ETHDataset class method) (src.dataset.gc_dataset.GCDataset class method) (src.dataset.orca_dataset.ORCADataset class method) (src.dataset.sdd_dataset.SDDDataset class method) (src.dataset.ucy_dataset.UCYDataset class method) (src.dataset.waymo_dataset.WayMoDataset class method)

load_data_batch() (src.dataset.eth_dataset.ETHDataset class method) (src.dataset.orca_dataset.ORCADataset class method) (src.dataset.sdd_dataset.SDDDataset class method) (src.dataset.ucy_dataset.UCYDataset class method) (src.dataset.waymo_dataset.WayMoDataset class method) load_font() (in module src.utils.plot)

M

map (src.dataset.base_dataset.RasterizedMap attribute), [1] map_data (src.tasks.simulate.SimulateState attribute) MeanPoolingLSTM (class in src.model.mean_pooling_lstm) merge_axes() (in module src.utils.plot) Model (class in src.model.model) module src src.dataset src.dataset.base_dataset src.dataset.eth_dataset src.dataset.gc_dataset src.dataset.orca_dataset src.dataset.sdd_dataset src.dataset.synthetic_dataset src.dataset.ucy_dataset src.dataset.waymo_dataset src.diffusion src.diffusion.ddim src.diffusion.ddpm src.model src.model.fourier_positional_encoding src.model.mean_pooling_lstm src.model.model src.model.multiscale_cnn src.model.nan_embedding src.model.new_model src.model.permuted src.model.relative_model src.model.residual src.model.sinusoidal_embedding src.tasks src.tasks.guidance src.tasks.sfm src.tasks.simulate src.tasks.test_once src.tasks.train_once src.tasks.visualize src.utils src.utils.auto_gpu src.utils.calc_xy_error src.utils.crop_axes_region src.utils.extract_patches src.utils.fix_parser src.utils.get_force_map src.utils.homography src.utils.json_compatible src.utils.logger src.utils.plot src.utils.seed src.utils.tag2ansi src.utils.timer src.utils.use_npu

MultiScaleCNN (class in src.model.multiscale_cnn)

N

NamedTimer (class in src.utils.timer) NanEmbedding (class in src.model.nan_embedding) NewModel (class in src.model.new_model)

noise_to_x0() (src.diffusion.ddpm.DDPM method) normalize_xy() (src.dataset.base_dataset.BaseDataset static method) npu_attention_fallback() (in module src.utils.use_npu) npu_attention_fallback_context() (in module src.utils.use_npu)

O

ORCADataset (class in src.dataset.orca_dataset)

P

ParallelTimer (class in src.utils.timer) ped_list (src.tasks.simulate.SimulateState attribute) Permuted (class in src.model.permuted)

plot_resilience() (in module src.utils.plot) plotOD() (in module src.utils.plot) pos_now (src.tasks.simulate.SimulateState attribute)

Q

query_free_memory() (src.utils.auto_gpu.AutoGPU method)

R

RasterizedMap (class in src.dataset.base_dataset) raw_fps (src.dataset.eth_dataset.ETHDataset attribute) (src.dataset.gc_dataset.GCDataset attribute) (src.dataset.orca_dataset.ORCADataset attribute) (src.dataset.sdd_dataset.SDDDataset attribute) (src.dataset.ucy_dataset.UCYDataset attribute) (src.dataset.waymo_dataset.WayMoDataset attribute)

recover_npu_attention() (in module src.utils.use_npu) RelativeModel (class in src.model.relative_model) resample_dataframe() (src.dataset.base_dataset.BaseDataset static method) Residual (class in src.model.residual)

S

save_cache() (src.dataset.base_dataset.BaseDataset static method) SDDDataset (class in src.dataset.sdd_dataset) seed_all() (in module src.utils.seed) set_map_embedding() (src.model.model.Model method) set_ped_embedding() (src.model.model.Model method) (src.model.new_model.NewModel method) (src.model.relative_model.RelativeModel method) set_sur_info() (src.model.model.Model method) (src.model.new_model.NewModel method) set_veh_embedding() (src.model.model.Model method) (src.model.relative_model.RelativeModel method) simulate_one_step() (in module src.tasks.simulate) SimulateState (class in src.tasks.simulate) SinusoidalEmbedding (class in src.model.sinusoidal_embedding) spd_now (src.tasks.simulate.SimulateState attribute) speed (src.utils.timer.Timer property) split_samples() (src.dataset.base_dataset.BaseDataset method) src module src.dataset module src.dataset.base_dataset module src.dataset.eth_dataset module src.dataset.gc_dataset module src.dataset.orca_dataset module src.dataset.sdd_dataset module src.dataset.synthetic_dataset module src.dataset.ucy_dataset module src.dataset.waymo_dataset module src.diffusion module src.diffusion.ddim module src.diffusion.ddpm module src.model module src.model.fourier_positional_encoding module src.model.mean_pooling_lstm module src.model.model module src.model.multiscale_cnn module src.model.nan_embedding module

src.model.new_model module src.model.permuted module src.model.relative_model module src.model.residual module src.model.sinusoidal_embedding module src.tasks module src.tasks.guidance module src.tasks.sfm module src.tasks.simulate module src.tasks.test_once module src.tasks.train_once module src.tasks.visualize module src.utils module src.utils.auto_gpu module src.utils.calc_xy_error module src.utils.crop_axes_region module src.utils.extract_patches module src.utils.fix_parser module src.utils.get_force_map module src.utils.homography module src.utils.json_compatible module src.utils.logger module src.utils.plot module src.utils.seed module src.utils.tag2ansi module src.utils.timer module src.utils.use_npu module

T

tag2ansi() (in module src.utils.tag2ansi) test_once() (in module src.tasks.test_once) time (src.utils.timer.NamedTimer property) (src.utils.timer.ParallelTimer property)

Timer (class in src.utils.timer) to() (src.diffusion.ddpm.DDPM method) total_time() (src.utils.timer.NamedTimer method) (src.utils.timer.ParallelTimer method) train_once() (in module src.tasks.train_once)

U

UCYDataset (class in src.dataset.ucy_dataset)

update_free_memory() (src.utils.auto_gpu.AutoGPU method) use_chinese_font() (in module src.utils.plot)

V

veh_list (src.tasks.simulate.SimulateState attribute) veh_now (src.tasks.simulate.SimulateState attribute)

vel_now (src.tasks.simulate.SimulateState attribute) visualize() (in module src.tasks.visualize)

W

WayMoDataset (class in src.dataset.waymo_dataset)

X

x0_to_noise() (src.diffusion.ddpm.DDPM method)

xmax (src.dataset.base_dataset.RasterizedMap attribute), [1] xmin (src.dataset.base_dataset.RasterizedMap attribute), [1]

Y

ymax (src.dataset.base_dataset.RasterizedMap attribute), [1]

ymin (src.dataset.base_dataset.RasterizedMap attribute), [1]