├── .gitignore ├── ReadME.md ├── commons.py ├── configs ├── main.yaml ├── mlp.yaml ├── pixel │ ├── cnn.yaml │ └── lin.yaml ├── train.yaml └── vqvae │ ├── cnn.yaml │ └── lin.yaml ├── datasets └── datasets2.py ├── gen_dist.py ├── gifs ├── comp.png ├── cv.gif ├── mlp.png ├── scenarios.png ├── vq.gif └── vqp.png ├── highway_env ├── __init__.py ├── envs │ ├── __init__.py │ ├── common │ │ ├── __init__.py │ │ ├── abstract.py │ │ ├── action.py │ │ ├── finite_mdp.py │ │ ├── graphics.py │ │ └── observation.py │ ├── exit_env.py │ ├── highway_env.py │ ├── intersection_env.py │ ├── lane_keeping_env.py │ ├── merge_env.py │ ├── parking_env.py │ ├── racetrack_env.py │ ├── roundabout_env.py │ ├── summon_env.py │ ├── two_way_env.py │ └── u_turn_env.py ├── interval.py ├── road │ ├── __init__.py │ ├── graphics.py │ ├── lane.py │ ├── regulation.py │ ├── road.py │ └── spline.py ├── utils.py └── vehicle │ ├── __init__.py │ ├── behavior.py │ ├── controller.py │ ├── dynamics.py │ ├── graphics.py │ ├── kinematics.py │ ├── objects.py │ ├── planning_controller.py │ └── uncertainty │ ├── __init__.py │ ├── estimation.py │ └── prediction.py ├── inference.py ├── inferenceJax.py ├── inferenceJax3.py ├── model ├── DOptimizer.py ├── DOptimizer2.py ├── DOptimizer3.py ├── DOptimizerJax.py ├── DOptimizerJax2.py ├── DOptimizerJax3.py ├── PixelCNN.py ├── PixelLinear.py └── VQVAE.py ├── requirements.txt ├── train_mlp.py ├── train_mlp2.py ├── train_pix.py ├── train_vq.py └── utils ├── bernstein.py ├── bernstein_jax.py ├── config.py ├── torchModel2Jax.py └── visualizations.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/.gitignore -------------------------------------------------------------------------------- /ReadME.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/ReadME.md -------------------------------------------------------------------------------- /commons.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/commons.py -------------------------------------------------------------------------------- /configs/main.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/configs/main.yaml -------------------------------------------------------------------------------- /configs/mlp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/configs/mlp.yaml -------------------------------------------------------------------------------- /configs/pixel/cnn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/configs/pixel/cnn.yaml -------------------------------------------------------------------------------- /configs/pixel/lin.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/configs/pixel/lin.yaml -------------------------------------------------------------------------------- /configs/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/configs/train.yaml -------------------------------------------------------------------------------- /configs/vqvae/cnn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/configs/vqvae/cnn.yaml -------------------------------------------------------------------------------- /configs/vqvae/lin.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/configs/vqvae/lin.yaml -------------------------------------------------------------------------------- /datasets/datasets2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/datasets/datasets2.py -------------------------------------------------------------------------------- /gen_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/gen_dist.py -------------------------------------------------------------------------------- /gifs/comp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/gifs/comp.png -------------------------------------------------------------------------------- /gifs/cv.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/gifs/cv.gif -------------------------------------------------------------------------------- /gifs/mlp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/gifs/mlp.png -------------------------------------------------------------------------------- /gifs/scenarios.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/gifs/scenarios.png -------------------------------------------------------------------------------- /gifs/vq.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/gifs/vq.gif -------------------------------------------------------------------------------- /gifs/vqp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/gifs/vqp.png -------------------------------------------------------------------------------- /highway_env/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/__init__.py -------------------------------------------------------------------------------- /highway_env/envs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/envs/__init__.py -------------------------------------------------------------------------------- /highway_env/envs/common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /highway_env/envs/common/abstract.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/envs/common/abstract.py -------------------------------------------------------------------------------- /highway_env/envs/common/action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/envs/common/action.py -------------------------------------------------------------------------------- /highway_env/envs/common/finite_mdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/envs/common/finite_mdp.py -------------------------------------------------------------------------------- /highway_env/envs/common/graphics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/envs/common/graphics.py -------------------------------------------------------------------------------- /highway_env/envs/common/observation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/envs/common/observation.py -------------------------------------------------------------------------------- /highway_env/envs/exit_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/envs/exit_env.py -------------------------------------------------------------------------------- /highway_env/envs/highway_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/envs/highway_env.py -------------------------------------------------------------------------------- /highway_env/envs/intersection_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/envs/intersection_env.py -------------------------------------------------------------------------------- /highway_env/envs/lane_keeping_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/envs/lane_keeping_env.py -------------------------------------------------------------------------------- /highway_env/envs/merge_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/envs/merge_env.py -------------------------------------------------------------------------------- /highway_env/envs/parking_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/envs/parking_env.py -------------------------------------------------------------------------------- /highway_env/envs/racetrack_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/envs/racetrack_env.py -------------------------------------------------------------------------------- /highway_env/envs/roundabout_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/envs/roundabout_env.py -------------------------------------------------------------------------------- /highway_env/envs/summon_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/envs/summon_env.py -------------------------------------------------------------------------------- /highway_env/envs/two_way_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/envs/two_way_env.py -------------------------------------------------------------------------------- /highway_env/envs/u_turn_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/envs/u_turn_env.py -------------------------------------------------------------------------------- /highway_env/interval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/interval.py -------------------------------------------------------------------------------- /highway_env/road/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /highway_env/road/graphics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/road/graphics.py -------------------------------------------------------------------------------- /highway_env/road/lane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/road/lane.py -------------------------------------------------------------------------------- /highway_env/road/regulation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/road/regulation.py -------------------------------------------------------------------------------- /highway_env/road/road.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/road/road.py -------------------------------------------------------------------------------- /highway_env/road/spline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/road/spline.py -------------------------------------------------------------------------------- /highway_env/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/utils.py -------------------------------------------------------------------------------- /highway_env/vehicle/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /highway_env/vehicle/behavior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/vehicle/behavior.py -------------------------------------------------------------------------------- /highway_env/vehicle/controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/vehicle/controller.py -------------------------------------------------------------------------------- /highway_env/vehicle/dynamics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/vehicle/dynamics.py -------------------------------------------------------------------------------- /highway_env/vehicle/graphics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/vehicle/graphics.py -------------------------------------------------------------------------------- /highway_env/vehicle/kinematics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/vehicle/kinematics.py -------------------------------------------------------------------------------- /highway_env/vehicle/objects.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/vehicle/objects.py -------------------------------------------------------------------------------- /highway_env/vehicle/planning_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/vehicle/planning_controller.py -------------------------------------------------------------------------------- /highway_env/vehicle/uncertainty/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /highway_env/vehicle/uncertainty/estimation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/vehicle/uncertainty/estimation.py -------------------------------------------------------------------------------- /highway_env/vehicle/uncertainty/prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/highway_env/vehicle/uncertainty/prediction.py -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/inference.py -------------------------------------------------------------------------------- /inferenceJax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/inferenceJax.py -------------------------------------------------------------------------------- /inferenceJax3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/inferenceJax3.py -------------------------------------------------------------------------------- /model/DOptimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/model/DOptimizer.py -------------------------------------------------------------------------------- /model/DOptimizer2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/model/DOptimizer2.py -------------------------------------------------------------------------------- /model/DOptimizer3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/model/DOptimizer3.py -------------------------------------------------------------------------------- /model/DOptimizerJax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/model/DOptimizerJax.py -------------------------------------------------------------------------------- /model/DOptimizerJax2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/model/DOptimizerJax2.py -------------------------------------------------------------------------------- /model/DOptimizerJax3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/model/DOptimizerJax3.py -------------------------------------------------------------------------------- /model/PixelCNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/model/PixelCNN.py -------------------------------------------------------------------------------- /model/PixelLinear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/model/PixelLinear.py -------------------------------------------------------------------------------- /model/VQVAE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/model/VQVAE.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/requirements.txt -------------------------------------------------------------------------------- /train_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/train_mlp.py -------------------------------------------------------------------------------- /train_mlp2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/train_mlp2.py -------------------------------------------------------------------------------- /train_pix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/train_pix.py -------------------------------------------------------------------------------- /train_vq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/train_vq.py -------------------------------------------------------------------------------- /utils/bernstein.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/utils/bernstein.py -------------------------------------------------------------------------------- /utils/bernstein_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/utils/bernstein_jax.py -------------------------------------------------------------------------------- /utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/utils/config.py -------------------------------------------------------------------------------- /utils/torchModel2Jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/utils/torchModel2Jax.py -------------------------------------------------------------------------------- /utils/visualizations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cisimon7/VQOptMain/HEAD/utils/visualizations.py --------------------------------------------------------------------------------