├── .gitignore ├── LICENSE ├── README.md ├── assets ├── architecture.jpg ├── bootstrap.min.css ├── corpus.js ├── editing.jpg ├── font.css ├── gpt_example.jpg ├── jquery.min.js ├── main_result.jpg ├── style.css └── teaser.jpg ├── cfgs ├── demo_inference.yaml ├── demo_train.yaml ├── inference.yaml └── train.yaml ├── data ├── demo │ └── llm_output │ │ └── debug.txt └── list │ ├── demo.txt │ ├── train.txt │ └── val.txt ├── index.html ├── lctgen ├── __init__.py ├── config │ ├── __init__.py │ ├── default.py │ └── path_cfg.py ├── core │ ├── basic.py │ └── registry.py ├── datasets │ ├── __init__.py │ ├── description.py │ ├── typedef.py │ ├── utils.py │ └── waymo_open_motion.py ├── gpt │ ├── __init__.py │ ├── cfgs │ │ ├── attr_ind_motion │ │ │ ├── non_api_cot.yaml │ │ │ ├── non_api_cot_attr.yaml │ │ │ └── non_api_cot_attr_20m.yaml │ │ └── debug.yaml │ ├── models.py │ ├── prompts │ │ ├── attr_ind_motion │ │ │ ├── backup.prompt │ │ │ ├── example.prompt │ │ │ ├── non_api_cot.prompt │ │ │ ├── non_api_cot_attr.prompt │ │ │ ├── non_api_cot_attr_20m.prompt │ │ │ ├── sys_non_api_cot.prompt │ │ │ ├── sys_non_api_cot_attr.prompt │ │ │ └── sys_non_api_cot_attr_20m.prompt │ │ └── editing │ │ │ └── edit.prompt │ └── test_text │ │ ├── attr.txt │ │ ├── attr_debug.txt │ │ ├── attr_label.txt │ │ ├── attr_result.txt │ │ ├── ciren.txt │ │ ├── ciren_debug.txt │ │ ├── ciren_label.txt │ │ └── ciren_result.txt ├── inference │ └── utils.py ├── lctgen │ └── demo │ │ └── config.yaml ├── main.py ├── metrics │ ├── __init__.py │ └── metric.py ├── models │ ├── __init__.py │ ├── base_model.py │ ├── blocks.py │ ├── detr_loss.py │ ├── detr_model.py │ ├── lctgen.py │ ├── loss.py │ ├── matcher.py │ ├── motion_utils.py │ ├── posprocess.py │ └── utils.py ├── train │ ├── 10-02-2023_20:51:28 │ │ └── version_0 │ │ │ ├── events.out.tfevents.1696297888.sota.1689256.0 │ │ │ └── hparams.yaml │ └── config.yaml └── trainer.py ├── requirements.txt ├── requirements_colab.txt ├── scripts └── process_all_data.py └── trafficgen └── utils ├── arg_parse.py ├── data_process ├── __init__.py ├── agent_process.py └── init_dataset.py ├── generate_gif.py ├── map_pb2.py ├── model_utils.py ├── nuplan.py ├── scenario_pb2.py ├── trans20.py ├── typedef.py ├── utils.py └── visual_init.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/README.md -------------------------------------------------------------------------------- /assets/architecture.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/assets/architecture.jpg -------------------------------------------------------------------------------- /assets/bootstrap.min.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/assets/bootstrap.min.css -------------------------------------------------------------------------------- /assets/corpus.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/assets/corpus.js -------------------------------------------------------------------------------- /assets/editing.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/assets/editing.jpg -------------------------------------------------------------------------------- /assets/font.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/assets/font.css -------------------------------------------------------------------------------- /assets/gpt_example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/assets/gpt_example.jpg -------------------------------------------------------------------------------- /assets/jquery.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/assets/jquery.min.js -------------------------------------------------------------------------------- /assets/main_result.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/assets/main_result.jpg -------------------------------------------------------------------------------- /assets/style.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/assets/style.css -------------------------------------------------------------------------------- /assets/teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/assets/teaser.jpg -------------------------------------------------------------------------------- /cfgs/demo_inference.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/cfgs/demo_inference.yaml -------------------------------------------------------------------------------- /cfgs/demo_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/cfgs/demo_train.yaml -------------------------------------------------------------------------------- /cfgs/inference.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/cfgs/inference.yaml -------------------------------------------------------------------------------- /cfgs/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/cfgs/train.yaml -------------------------------------------------------------------------------- /data/demo/llm_output/debug.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/data/demo/llm_output/debug.txt -------------------------------------------------------------------------------- /data/list/demo.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/data/list/demo.txt -------------------------------------------------------------------------------- /data/list/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/data/list/train.txt -------------------------------------------------------------------------------- /data/list/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/data/list/val.txt -------------------------------------------------------------------------------- /index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/index.html -------------------------------------------------------------------------------- /lctgen/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/__init__.py -------------------------------------------------------------------------------- /lctgen/config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lctgen/config/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/config/default.py -------------------------------------------------------------------------------- /lctgen/config/path_cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/config/path_cfg.py -------------------------------------------------------------------------------- /lctgen/core/basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/core/basic.py -------------------------------------------------------------------------------- /lctgen/core/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/core/registry.py -------------------------------------------------------------------------------- /lctgen/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | from .waymo_open_motion import * -------------------------------------------------------------------------------- /lctgen/datasets/description.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/datasets/description.py -------------------------------------------------------------------------------- /lctgen/datasets/typedef.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/datasets/typedef.py -------------------------------------------------------------------------------- /lctgen/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/datasets/utils.py -------------------------------------------------------------------------------- /lctgen/datasets/waymo_open_motion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/datasets/waymo_open_motion.py -------------------------------------------------------------------------------- /lctgen/gpt/__init__.py: -------------------------------------------------------------------------------- 1 | from .models import * -------------------------------------------------------------------------------- /lctgen/gpt/cfgs/attr_ind_motion/non_api_cot.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/gpt/cfgs/attr_ind_motion/non_api_cot.yaml -------------------------------------------------------------------------------- /lctgen/gpt/cfgs/attr_ind_motion/non_api_cot_attr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/gpt/cfgs/attr_ind_motion/non_api_cot_attr.yaml -------------------------------------------------------------------------------- /lctgen/gpt/cfgs/attr_ind_motion/non_api_cot_attr_20m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/gpt/cfgs/attr_ind_motion/non_api_cot_attr_20m.yaml -------------------------------------------------------------------------------- /lctgen/gpt/cfgs/debug.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/gpt/cfgs/debug.yaml -------------------------------------------------------------------------------- /lctgen/gpt/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/gpt/models.py -------------------------------------------------------------------------------- /lctgen/gpt/prompts/attr_ind_motion/backup.prompt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/gpt/prompts/attr_ind_motion/backup.prompt -------------------------------------------------------------------------------- /lctgen/gpt/prompts/attr_ind_motion/example.prompt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/gpt/prompts/attr_ind_motion/example.prompt -------------------------------------------------------------------------------- /lctgen/gpt/prompts/attr_ind_motion/non_api_cot.prompt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/gpt/prompts/attr_ind_motion/non_api_cot.prompt -------------------------------------------------------------------------------- /lctgen/gpt/prompts/attr_ind_motion/non_api_cot_attr.prompt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/gpt/prompts/attr_ind_motion/non_api_cot_attr.prompt -------------------------------------------------------------------------------- /lctgen/gpt/prompts/attr_ind_motion/non_api_cot_attr_20m.prompt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/gpt/prompts/attr_ind_motion/non_api_cot_attr_20m.prompt -------------------------------------------------------------------------------- /lctgen/gpt/prompts/attr_ind_motion/sys_non_api_cot.prompt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/gpt/prompts/attr_ind_motion/sys_non_api_cot.prompt -------------------------------------------------------------------------------- /lctgen/gpt/prompts/attr_ind_motion/sys_non_api_cot_attr.prompt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/gpt/prompts/attr_ind_motion/sys_non_api_cot_attr.prompt -------------------------------------------------------------------------------- /lctgen/gpt/prompts/attr_ind_motion/sys_non_api_cot_attr_20m.prompt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/gpt/prompts/attr_ind_motion/sys_non_api_cot_attr_20m.prompt -------------------------------------------------------------------------------- /lctgen/gpt/prompts/editing/edit.prompt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/gpt/prompts/editing/edit.prompt -------------------------------------------------------------------------------- /lctgen/gpt/test_text/attr.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/gpt/test_text/attr.txt -------------------------------------------------------------------------------- /lctgen/gpt/test_text/attr_debug.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/gpt/test_text/attr_debug.txt -------------------------------------------------------------------------------- /lctgen/gpt/test_text/attr_label.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/gpt/test_text/attr_label.txt -------------------------------------------------------------------------------- /lctgen/gpt/test_text/attr_result.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/gpt/test_text/attr_result.txt -------------------------------------------------------------------------------- /lctgen/gpt/test_text/ciren.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/gpt/test_text/ciren.txt -------------------------------------------------------------------------------- /lctgen/gpt/test_text/ciren_debug.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/gpt/test_text/ciren_debug.txt -------------------------------------------------------------------------------- /lctgen/gpt/test_text/ciren_label.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/gpt/test_text/ciren_label.txt -------------------------------------------------------------------------------- /lctgen/gpt/test_text/ciren_result.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/gpt/test_text/ciren_result.txt -------------------------------------------------------------------------------- /lctgen/inference/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/inference/utils.py -------------------------------------------------------------------------------- /lctgen/lctgen/demo/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/lctgen/demo/config.yaml -------------------------------------------------------------------------------- /lctgen/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/main.py -------------------------------------------------------------------------------- /lctgen/metrics/__init__.py: -------------------------------------------------------------------------------- 1 | from .metric import * -------------------------------------------------------------------------------- /lctgen/metrics/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/metrics/metric.py -------------------------------------------------------------------------------- /lctgen/models/__init__.py: -------------------------------------------------------------------------------- 1 | from .lctgen import * -------------------------------------------------------------------------------- /lctgen/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/models/base_model.py -------------------------------------------------------------------------------- /lctgen/models/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/models/blocks.py -------------------------------------------------------------------------------- /lctgen/models/detr_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/models/detr_loss.py -------------------------------------------------------------------------------- /lctgen/models/detr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/models/detr_model.py -------------------------------------------------------------------------------- /lctgen/models/lctgen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/models/lctgen.py -------------------------------------------------------------------------------- /lctgen/models/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/models/loss.py -------------------------------------------------------------------------------- /lctgen/models/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/models/matcher.py -------------------------------------------------------------------------------- /lctgen/models/motion_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/models/motion_utils.py -------------------------------------------------------------------------------- /lctgen/models/posprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/models/posprocess.py -------------------------------------------------------------------------------- /lctgen/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/models/utils.py -------------------------------------------------------------------------------- /lctgen/train/10-02-2023_20:51:28/version_0/events.out.tfevents.1696297888.sota.1689256.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/train/10-02-2023_20:51:28/version_0/events.out.tfevents.1696297888.sota.1689256.0 -------------------------------------------------------------------------------- /lctgen/train/10-02-2023_20:51:28/version_0/hparams.yaml: -------------------------------------------------------------------------------- 1 | {} 2 | -------------------------------------------------------------------------------- /lctgen/train/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/train/config.yaml -------------------------------------------------------------------------------- /lctgen/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/lctgen/trainer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/requirements.txt -------------------------------------------------------------------------------- /requirements_colab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/requirements_colab.txt -------------------------------------------------------------------------------- /scripts/process_all_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/scripts/process_all_data.py -------------------------------------------------------------------------------- /trafficgen/utils/arg_parse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/trafficgen/utils/arg_parse.py -------------------------------------------------------------------------------- /trafficgen/utils/data_process/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /trafficgen/utils/data_process/agent_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/trafficgen/utils/data_process/agent_process.py -------------------------------------------------------------------------------- /trafficgen/utils/data_process/init_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/trafficgen/utils/data_process/init_dataset.py -------------------------------------------------------------------------------- /trafficgen/utils/generate_gif.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/trafficgen/utils/generate_gif.py -------------------------------------------------------------------------------- /trafficgen/utils/map_pb2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/trafficgen/utils/map_pb2.py -------------------------------------------------------------------------------- /trafficgen/utils/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/trafficgen/utils/model_utils.py -------------------------------------------------------------------------------- /trafficgen/utils/nuplan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/trafficgen/utils/nuplan.py -------------------------------------------------------------------------------- /trafficgen/utils/scenario_pb2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/trafficgen/utils/scenario_pb2.py -------------------------------------------------------------------------------- /trafficgen/utils/trans20.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/trafficgen/utils/trans20.py -------------------------------------------------------------------------------- /trafficgen/utils/typedef.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/trafficgen/utils/typedef.py -------------------------------------------------------------------------------- /trafficgen/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/trafficgen/utils/utils.py -------------------------------------------------------------------------------- /trafficgen/utils/visual_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ariostgx/lctgen/HEAD/trafficgen/utils/visual_init.py --------------------------------------------------------------------------------