├── .gitignore ├── LICENSE ├── README.md ├── collect_data.py ├── configs ├── __init__.py ├── config.py └── driving_config.py ├── driving_sim ├── __init__.py ├── actions │ ├── __init__.py │ ├── action.py │ ├── trajectory_accel_action.py │ └── xy_accel_action.py ├── car.py ├── constants.py ├── drivers │ ├── __init__.py │ ├── driver.py │ └── oned_drivers.py ├── envs │ ├── __init__.py │ ├── t_intersection.py │ ├── t_intersection_pred_front.py │ ├── t_intersection_pred_front_action.py │ └── traffic_env.py ├── rendering.py ├── road.py ├── utils │ ├── info.py │ └── trajectory.py └── vec_env │ ├── __init__.py │ ├── dummy_vec_env.py │ ├── envs.py │ ├── logger.py │ ├── running_mean_std.py │ ├── shmem_vec_env.py │ ├── tile_images.py │ ├── util.py │ ├── vec_env.py │ ├── vec_frame_stack.py │ ├── vec_normalize.py │ ├── vec_pretext_normalize.py │ └── vec_remove_dict_obs.py ├── figures ├── latent_space_baseline.png ├── latent_space_ours.png ├── opening.png └── ours40.gif ├── plot_pretext.py ├── plot_rl.py ├── pretext ├── __init__.py ├── data_collector.py ├── data_loader.py ├── loss.py └── pretext_models │ ├── __init__.py │ ├── cvae_model.py │ ├── decoders.py │ └── pretext_rnn_base.py ├── requirements.txt ├── rl ├── .gitignore ├── __init__.py ├── distributions.py ├── envs.py ├── evaluation.py ├── ppo │ ├── __init__.py │ └── ppo.py ├── rl_models │ ├── lstm_attn.py │ ├── lstm_no_attn.py │ ├── policy.py │ └── rnn_base.py ├── shmem_vec_env.py ├── storage.py └── utils.py ├── test_pretext.py ├── test_rl.py ├── train_pretext.py ├── train_rl.py └── trained_models ├── pretext ├── public_morton │ ├── checkpoints │ │ └── 995.pt │ ├── configs │ │ ├── __init__.py │ │ ├── config.py │ │ └── driving_config.py │ └── progress.csv └── public_ours │ ├── checkpoints │ └── 995.pt │ ├── configs │ ├── __init__.py │ ├── config.py │ └── driving_config.py │ └── progress.csv └── rl └── con40 ├── public_morton_rl ├── checkpoints │ └── 25200.pt ├── configs │ ├── __init__.py │ ├── config.py │ └── driving_config.py ├── progress.csv └── test │ └── test_25200.pt.log └── public_ours_rl ├── .~lock.progress.csv# ├── checkpoints └── 26800.pt ├── configs ├── __init__.py ├── config.py └── driving_config.py ├── progress.csv └── test └── test_26800.pt.log /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/README.md -------------------------------------------------------------------------------- /collect_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/collect_data.py -------------------------------------------------------------------------------- /configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/configs/config.py -------------------------------------------------------------------------------- /configs/driving_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/configs/driving_config.py -------------------------------------------------------------------------------- /driving_sim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/__init__.py -------------------------------------------------------------------------------- /driving_sim/actions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /driving_sim/actions/action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/actions/action.py -------------------------------------------------------------------------------- /driving_sim/actions/trajectory_accel_action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/actions/trajectory_accel_action.py -------------------------------------------------------------------------------- /driving_sim/actions/xy_accel_action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/actions/xy_accel_action.py -------------------------------------------------------------------------------- /driving_sim/car.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/car.py -------------------------------------------------------------------------------- /driving_sim/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/constants.py -------------------------------------------------------------------------------- /driving_sim/drivers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /driving_sim/drivers/driver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/drivers/driver.py -------------------------------------------------------------------------------- /driving_sim/drivers/oned_drivers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/drivers/oned_drivers.py -------------------------------------------------------------------------------- /driving_sim/envs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/envs/__init__.py -------------------------------------------------------------------------------- /driving_sim/envs/t_intersection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/envs/t_intersection.py -------------------------------------------------------------------------------- /driving_sim/envs/t_intersection_pred_front.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/envs/t_intersection_pred_front.py -------------------------------------------------------------------------------- /driving_sim/envs/t_intersection_pred_front_action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/envs/t_intersection_pred_front_action.py -------------------------------------------------------------------------------- /driving_sim/envs/traffic_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/envs/traffic_env.py -------------------------------------------------------------------------------- /driving_sim/rendering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/rendering.py -------------------------------------------------------------------------------- /driving_sim/road.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/road.py -------------------------------------------------------------------------------- /driving_sim/utils/info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/utils/info.py -------------------------------------------------------------------------------- /driving_sim/utils/trajectory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/utils/trajectory.py -------------------------------------------------------------------------------- /driving_sim/vec_env/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/vec_env/__init__.py -------------------------------------------------------------------------------- /driving_sim/vec_env/dummy_vec_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/vec_env/dummy_vec_env.py -------------------------------------------------------------------------------- /driving_sim/vec_env/envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/vec_env/envs.py -------------------------------------------------------------------------------- /driving_sim/vec_env/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/vec_env/logger.py -------------------------------------------------------------------------------- /driving_sim/vec_env/running_mean_std.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/vec_env/running_mean_std.py -------------------------------------------------------------------------------- /driving_sim/vec_env/shmem_vec_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/vec_env/shmem_vec_env.py -------------------------------------------------------------------------------- /driving_sim/vec_env/tile_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/vec_env/tile_images.py -------------------------------------------------------------------------------- /driving_sim/vec_env/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/vec_env/util.py -------------------------------------------------------------------------------- /driving_sim/vec_env/vec_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/vec_env/vec_env.py -------------------------------------------------------------------------------- /driving_sim/vec_env/vec_frame_stack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/vec_env/vec_frame_stack.py -------------------------------------------------------------------------------- /driving_sim/vec_env/vec_normalize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/vec_env/vec_normalize.py -------------------------------------------------------------------------------- /driving_sim/vec_env/vec_pretext_normalize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/vec_env/vec_pretext_normalize.py -------------------------------------------------------------------------------- /driving_sim/vec_env/vec_remove_dict_obs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/driving_sim/vec_env/vec_remove_dict_obs.py -------------------------------------------------------------------------------- /figures/latent_space_baseline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/figures/latent_space_baseline.png -------------------------------------------------------------------------------- /figures/latent_space_ours.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/figures/latent_space_ours.png -------------------------------------------------------------------------------- /figures/opening.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/figures/opening.png -------------------------------------------------------------------------------- /figures/ours40.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/figures/ours40.gif -------------------------------------------------------------------------------- /plot_pretext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/plot_pretext.py -------------------------------------------------------------------------------- /plot_rl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/plot_rl.py -------------------------------------------------------------------------------- /pretext/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /pretext/data_collector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/pretext/data_collector.py -------------------------------------------------------------------------------- /pretext/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/pretext/data_loader.py -------------------------------------------------------------------------------- /pretext/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/pretext/loss.py -------------------------------------------------------------------------------- /pretext/pretext_models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretext/pretext_models/cvae_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/pretext/pretext_models/cvae_model.py -------------------------------------------------------------------------------- /pretext/pretext_models/decoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/pretext/pretext_models/decoders.py -------------------------------------------------------------------------------- /pretext/pretext_models/pretext_rnn_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/pretext/pretext_models/pretext_rnn_base.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/requirements.txt -------------------------------------------------------------------------------- /rl/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/rl/.gitignore -------------------------------------------------------------------------------- /rl/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /rl/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/rl/distributions.py -------------------------------------------------------------------------------- /rl/envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/rl/envs.py -------------------------------------------------------------------------------- /rl/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/rl/evaluation.py -------------------------------------------------------------------------------- /rl/ppo/__init__.py: -------------------------------------------------------------------------------- 1 | from .ppo import PPO -------------------------------------------------------------------------------- /rl/ppo/ppo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/rl/ppo/ppo.py -------------------------------------------------------------------------------- /rl/rl_models/lstm_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/rl/rl_models/lstm_attn.py -------------------------------------------------------------------------------- /rl/rl_models/lstm_no_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/rl/rl_models/lstm_no_attn.py -------------------------------------------------------------------------------- /rl/rl_models/policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/rl/rl_models/policy.py -------------------------------------------------------------------------------- /rl/rl_models/rnn_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/rl/rl_models/rnn_base.py -------------------------------------------------------------------------------- /rl/shmem_vec_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/rl/shmem_vec_env.py -------------------------------------------------------------------------------- /rl/storage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/rl/storage.py -------------------------------------------------------------------------------- /rl/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/rl/utils.py -------------------------------------------------------------------------------- /test_pretext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/test_pretext.py -------------------------------------------------------------------------------- /test_rl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/test_rl.py -------------------------------------------------------------------------------- /train_pretext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/train_pretext.py -------------------------------------------------------------------------------- /train_rl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/train_rl.py -------------------------------------------------------------------------------- /trained_models/pretext/public_morton/checkpoints/995.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/trained_models/pretext/public_morton/checkpoints/995.pt -------------------------------------------------------------------------------- /trained_models/pretext/public_morton/configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /trained_models/pretext/public_morton/configs/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/trained_models/pretext/public_morton/configs/config.py -------------------------------------------------------------------------------- /trained_models/pretext/public_morton/configs/driving_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/trained_models/pretext/public_morton/configs/driving_config.py -------------------------------------------------------------------------------- /trained_models/pretext/public_morton/progress.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/trained_models/pretext/public_morton/progress.csv -------------------------------------------------------------------------------- /trained_models/pretext/public_ours/checkpoints/995.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/trained_models/pretext/public_ours/checkpoints/995.pt -------------------------------------------------------------------------------- /trained_models/pretext/public_ours/configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /trained_models/pretext/public_ours/configs/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/trained_models/pretext/public_ours/configs/config.py -------------------------------------------------------------------------------- /trained_models/pretext/public_ours/configs/driving_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/trained_models/pretext/public_ours/configs/driving_config.py -------------------------------------------------------------------------------- /trained_models/pretext/public_ours/progress.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/trained_models/pretext/public_ours/progress.csv -------------------------------------------------------------------------------- /trained_models/rl/con40/public_morton_rl/checkpoints/25200.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/trained_models/rl/con40/public_morton_rl/checkpoints/25200.pt -------------------------------------------------------------------------------- /trained_models/rl/con40/public_morton_rl/configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /trained_models/rl/con40/public_morton_rl/configs/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/trained_models/rl/con40/public_morton_rl/configs/config.py -------------------------------------------------------------------------------- /trained_models/rl/con40/public_morton_rl/configs/driving_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/trained_models/rl/con40/public_morton_rl/configs/driving_config.py -------------------------------------------------------------------------------- /trained_models/rl/con40/public_morton_rl/progress.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/trained_models/rl/con40/public_morton_rl/progress.csv -------------------------------------------------------------------------------- /trained_models/rl/con40/public_morton_rl/test/test_25200.pt.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/trained_models/rl/con40/public_morton_rl/test/test_25200.pt.log -------------------------------------------------------------------------------- /trained_models/rl/con40/public_ours_rl/.~lock.progress.csv#: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/trained_models/rl/con40/public_ours_rl/.~lock.progress.csv# -------------------------------------------------------------------------------- /trained_models/rl/con40/public_ours_rl/checkpoints/26800.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/trained_models/rl/con40/public_ours_rl/checkpoints/26800.pt -------------------------------------------------------------------------------- /trained_models/rl/con40/public_ours_rl/configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /trained_models/rl/con40/public_ours_rl/configs/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/trained_models/rl/con40/public_ours_rl/configs/config.py -------------------------------------------------------------------------------- /trained_models/rl/con40/public_ours_rl/configs/driving_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/trained_models/rl/con40/public_ours_rl/configs/driving_config.py -------------------------------------------------------------------------------- /trained_models/rl/con40/public_ours_rl/progress.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/trained_models/rl/con40/public_ours_rl/progress.csv -------------------------------------------------------------------------------- /trained_models/rl/con40/public_ours_rl/test/test_26800.pt.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shuijing725/VAE_trait_inference/HEAD/trained_models/rl/con40/public_ours_rl/test/test_26800.pt.log --------------------------------------------------------------------------------