├── LICENSE ├── README.md ├── assets ├── gif │ ├── interplan │ │ ├── car_crash.gif │ │ ├── jaywalk.gif │ │ └── nudge.gif │ └── nuplan │ │ ├── cutin.gif │ │ ├── ped.gif │ │ └── right_turn.gif └── imgs │ ├── cases_img │ ├── case17.png │ ├── case2.png │ ├── case5.png │ ├── case8.png │ └── case9.png │ └── framework.png ├── flow_planner ├── core │ ├── __init__.py │ ├── common │ │ ├── json_utils.py │ │ ├── model_summary.py │ │ └── train_scheduler.py │ └── flow_matching_core.py ├── data │ ├── __init__.py │ ├── augmentation │ │ ├── __init__.py │ │ ├── nuplan_augmentation.py │ │ └── state_aug.py │ ├── data_process │ │ ├── agent_process.py │ │ ├── data_processor.py │ │ ├── map_process.py │ │ ├── roadblock_utils.py │ │ └── utils.py │ ├── dataset │ │ ├── __init__.py │ │ └── nuplan.py │ ├── normalization │ │ ├── __init__.py │ │ ├── obs_normalize.py │ │ └── state_normalize.py │ └── utils │ │ ├── __init__.py │ │ ├── collect.py │ │ └── masked_tensor.py ├── model │ ├── __init__.py │ ├── flow_planner_model │ │ ├── decoder.py │ │ ├── encoder.py │ │ ├── flow_planner.py │ │ ├── flow_utils │ │ │ ├── flow_ode.py │ │ │ └── velocity_model.py │ │ └── global_attention.py │ ├── model_base.py │ ├── model_utils │ │ ├── __init__.py │ │ ├── input_preprocess.py │ │ ├── time_sampler.py │ │ ├── tool_func.py │ │ └── traj_tool.py │ └── modules │ │ ├── decoder_modules.py │ │ └── encoder_modules.py ├── nuplan_simulation │ ├── __init__.py │ ├── planner │ │ ├── __init__.py │ │ └── flow_planner.yaml │ └── scenario_filter │ │ ├── __init__.py │ │ ├── test14-hard.yaml │ │ ├── test14-random.yaml │ │ └── val14.yaml ├── planner.py ├── recorder │ ├── __init__.py │ └── tensorboard.py ├── run_script │ ├── launch_sim_nuplan.sh │ └── launch_train.sh ├── script │ ├── __init__.py │ ├── core │ │ └── flow_matching.yaml │ ├── data │ │ ├── __init__.py │ │ ├── dataset │ │ │ └── nuplan_data.yaml │ │ └── input_aug │ │ │ └── input_aug.yaml │ ├── ema │ │ └── ema.yaml │ ├── flow_planner_standard.yaml │ ├── model │ │ ├── flow_planner.yaml │ │ ├── obs_normalizer │ │ │ └── nuplan_obs_normalizer.yaml │ │ └── state_normalizer │ │ │ └── nuplan_state_normalizer.yaml │ ├── normalization_stats │ │ └── waypoints_norm_stats.yaml │ ├── optimizer │ │ └── adamw.yaml │ ├── recorder │ │ └── tensorboard.yaml │ └── scheduler │ │ ├── exponential_lr.yaml │ │ ├── step_lr.yaml │ │ └── warmup_const.yaml ├── train_utils │ ├── ddp.py │ └── save_model.py └── trainer.py ├── requirements.txt └── setup.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/README.md -------------------------------------------------------------------------------- /assets/gif/interplan/car_crash.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/assets/gif/interplan/car_crash.gif -------------------------------------------------------------------------------- /assets/gif/interplan/jaywalk.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/assets/gif/interplan/jaywalk.gif -------------------------------------------------------------------------------- /assets/gif/interplan/nudge.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/assets/gif/interplan/nudge.gif -------------------------------------------------------------------------------- /assets/gif/nuplan/cutin.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/assets/gif/nuplan/cutin.gif -------------------------------------------------------------------------------- /assets/gif/nuplan/ped.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/assets/gif/nuplan/ped.gif -------------------------------------------------------------------------------- /assets/gif/nuplan/right_turn.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/assets/gif/nuplan/right_turn.gif -------------------------------------------------------------------------------- /assets/imgs/cases_img/case17.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/assets/imgs/cases_img/case17.png -------------------------------------------------------------------------------- /assets/imgs/cases_img/case2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/assets/imgs/cases_img/case2.png -------------------------------------------------------------------------------- /assets/imgs/cases_img/case5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/assets/imgs/cases_img/case5.png -------------------------------------------------------------------------------- /assets/imgs/cases_img/case8.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/assets/imgs/cases_img/case8.png -------------------------------------------------------------------------------- /assets/imgs/cases_img/case9.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/assets/imgs/cases_img/case9.png -------------------------------------------------------------------------------- /assets/imgs/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/assets/imgs/framework.png -------------------------------------------------------------------------------- /flow_planner/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/core/__init__.py -------------------------------------------------------------------------------- /flow_planner/core/common/json_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/core/common/json_utils.py -------------------------------------------------------------------------------- /flow_planner/core/common/model_summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/core/common/model_summary.py -------------------------------------------------------------------------------- /flow_planner/core/common/train_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/core/common/train_scheduler.py -------------------------------------------------------------------------------- /flow_planner/core/flow_matching_core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/core/flow_matching_core.py -------------------------------------------------------------------------------- /flow_planner/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/data/__init__.py -------------------------------------------------------------------------------- /flow_planner/data/augmentation/__init__.py: -------------------------------------------------------------------------------- 1 | # -------------------------------------------------------------------------------- /flow_planner/data/augmentation/nuplan_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/data/augmentation/nuplan_augmentation.py -------------------------------------------------------------------------------- /flow_planner/data/augmentation/state_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/data/augmentation/state_aug.py -------------------------------------------------------------------------------- /flow_planner/data/data_process/agent_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/data/data_process/agent_process.py -------------------------------------------------------------------------------- /flow_planner/data/data_process/data_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/data/data_process/data_processor.py -------------------------------------------------------------------------------- /flow_planner/data/data_process/map_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/data/data_process/map_process.py -------------------------------------------------------------------------------- /flow_planner/data/data_process/roadblock_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/data/data_process/roadblock_utils.py -------------------------------------------------------------------------------- /flow_planner/data/data_process/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/data/data_process/utils.py -------------------------------------------------------------------------------- /flow_planner/data/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | # -------------------------------------------------------------------------------- /flow_planner/data/dataset/nuplan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/data/dataset/nuplan.py -------------------------------------------------------------------------------- /flow_planner/data/normalization/__init__.py: -------------------------------------------------------------------------------- 1 | # -------------------------------------------------------------------------------- /flow_planner/data/normalization/obs_normalize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/data/normalization/obs_normalize.py -------------------------------------------------------------------------------- /flow_planner/data/normalization/state_normalize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/data/normalization/state_normalize.py -------------------------------------------------------------------------------- /flow_planner/data/utils/__init__.py: -------------------------------------------------------------------------------- 1 | # -------------------------------------------------------------------------------- /flow_planner/data/utils/collect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/data/utils/collect.py -------------------------------------------------------------------------------- /flow_planner/data/utils/masked_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/data/utils/masked_tensor.py -------------------------------------------------------------------------------- /flow_planner/model/__init__.py: -------------------------------------------------------------------------------- 1 | # -------------------------------------------------------------------------------- /flow_planner/model/flow_planner_model/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/model/flow_planner_model/decoder.py -------------------------------------------------------------------------------- /flow_planner/model/flow_planner_model/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/model/flow_planner_model/encoder.py -------------------------------------------------------------------------------- /flow_planner/model/flow_planner_model/flow_planner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/model/flow_planner_model/flow_planner.py -------------------------------------------------------------------------------- /flow_planner/model/flow_planner_model/flow_utils/flow_ode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/model/flow_planner_model/flow_utils/flow_ode.py -------------------------------------------------------------------------------- /flow_planner/model/flow_planner_model/flow_utils/velocity_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/model/flow_planner_model/flow_utils/velocity_model.py -------------------------------------------------------------------------------- /flow_planner/model/flow_planner_model/global_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/model/flow_planner_model/global_attention.py -------------------------------------------------------------------------------- /flow_planner/model/model_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/model/model_base.py -------------------------------------------------------------------------------- /flow_planner/model/model_utils/__init__.py: -------------------------------------------------------------------------------- 1 | # -------------------------------------------------------------------------------- /flow_planner/model/model_utils/input_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/model/model_utils/input_preprocess.py -------------------------------------------------------------------------------- /flow_planner/model/model_utils/time_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/model/model_utils/time_sampler.py -------------------------------------------------------------------------------- /flow_planner/model/model_utils/tool_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/model/model_utils/tool_func.py -------------------------------------------------------------------------------- /flow_planner/model/model_utils/traj_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/model/model_utils/traj_tool.py -------------------------------------------------------------------------------- /flow_planner/model/modules/decoder_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/model/modules/decoder_modules.py -------------------------------------------------------------------------------- /flow_planner/model/modules/encoder_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/model/modules/encoder_modules.py -------------------------------------------------------------------------------- /flow_planner/nuplan_simulation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /flow_planner/nuplan_simulation/planner/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /flow_planner/nuplan_simulation/planner/flow_planner.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/nuplan_simulation/planner/flow_planner.yaml -------------------------------------------------------------------------------- /flow_planner/nuplan_simulation/scenario_filter/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /flow_planner/nuplan_simulation/scenario_filter/test14-hard.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/nuplan_simulation/scenario_filter/test14-hard.yaml -------------------------------------------------------------------------------- /flow_planner/nuplan_simulation/scenario_filter/test14-random.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/nuplan_simulation/scenario_filter/test14-random.yaml -------------------------------------------------------------------------------- /flow_planner/nuplan_simulation/scenario_filter/val14.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/nuplan_simulation/scenario_filter/val14.yaml -------------------------------------------------------------------------------- /flow_planner/planner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/planner.py -------------------------------------------------------------------------------- /flow_planner/recorder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/recorder/__init__.py -------------------------------------------------------------------------------- /flow_planner/recorder/tensorboard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/recorder/tensorboard.py -------------------------------------------------------------------------------- /flow_planner/run_script/launch_sim_nuplan.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/run_script/launch_sim_nuplan.sh -------------------------------------------------------------------------------- /flow_planner/run_script/launch_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/run_script/launch_train.sh -------------------------------------------------------------------------------- /flow_planner/script/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /flow_planner/script/core/flow_matching.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/script/core/flow_matching.yaml -------------------------------------------------------------------------------- /flow_planner/script/data/__init__.py: -------------------------------------------------------------------------------- 1 | # -------------------------------------------------------------------------------- /flow_planner/script/data/dataset/nuplan_data.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/script/data/dataset/nuplan_data.yaml -------------------------------------------------------------------------------- /flow_planner/script/data/input_aug/input_aug.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/script/data/input_aug/input_aug.yaml -------------------------------------------------------------------------------- /flow_planner/script/ema/ema.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/script/ema/ema.yaml -------------------------------------------------------------------------------- /flow_planner/script/flow_planner_standard.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/script/flow_planner_standard.yaml -------------------------------------------------------------------------------- /flow_planner/script/model/flow_planner.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/script/model/flow_planner.yaml -------------------------------------------------------------------------------- /flow_planner/script/model/obs_normalizer/nuplan_obs_normalizer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/script/model/obs_normalizer/nuplan_obs_normalizer.yaml -------------------------------------------------------------------------------- /flow_planner/script/model/state_normalizer/nuplan_state_normalizer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/script/model/state_normalizer/nuplan_state_normalizer.yaml -------------------------------------------------------------------------------- /flow_planner/script/normalization_stats/waypoints_norm_stats.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/script/normalization_stats/waypoints_norm_stats.yaml -------------------------------------------------------------------------------- /flow_planner/script/optimizer/adamw.yaml: -------------------------------------------------------------------------------- 1 | _target_: torch.optim.AdamW 2 | _convert_: all 3 | lr: 1e-4 -------------------------------------------------------------------------------- /flow_planner/script/recorder/tensorboard.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/script/recorder/tensorboard.yaml -------------------------------------------------------------------------------- /flow_planner/script/scheduler/exponential_lr.yaml: -------------------------------------------------------------------------------- 1 | _target_: torch.optim.lr_scheduler.ExponentialLR 2 | gamma: 0.9 -------------------------------------------------------------------------------- /flow_planner/script/scheduler/step_lr.yaml: -------------------------------------------------------------------------------- 1 | _target_: torch.optim.lr_scheduler.StepLR 2 | step_size: 5 3 | gamma: 0.1 -------------------------------------------------------------------------------- /flow_planner/script/scheduler/warmup_const.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/script/scheduler/warmup_const.yaml -------------------------------------------------------------------------------- /flow_planner/train_utils/ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/train_utils/ddp.py -------------------------------------------------------------------------------- /flow_planner/train_utils/save_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/train_utils/save_model.py -------------------------------------------------------------------------------- /flow_planner/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/flow_planner/trainer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DiffusionAD/Flow-Planner/HEAD/setup.py --------------------------------------------------------------------------------