├── .github └── workflows │ └── ci.yml ├── .gitignore ├── README.md ├── benchmarks ├── __init__.py ├── benchmark_performance.py └── benchmark_scaling.py ├── cbs ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-39.pyc │ └── a_star.cpython-39.pyc ├── a_star.py ├── cbs.py ├── examples │ ├── input.yaml │ ├── output.yaml │ ├── test.yaml │ └── test_2.yaml └── visualize.py ├── configs ├── config_baseline.yaml └── config_gnn.yaml ├── data_generation ├── __init__.py ├── dataset_gen.py ├── main_data.py ├── record.py └── trajectory_parser.py ├── data_loader.py ├── demos ├── README.md └── cbs_env_demo.py ├── evaluate.py ├── example.gif ├── example.py ├── grid ├── __init__.py ├── controller_grid.py ├── env_graph_grid.py └── env_graph_gridv1.py ├── models ├── __init__.py ├── framework_baseline.py ├── framework_gnn.py ├── framework_gnn_message.py └── networks │ ├── __init__.py │ ├── gnn.py │ └── utils_weights.py ├── requirements.txt ├── tests ├── __init__.py ├── test_cbs.py ├── test_cbs_performance.py ├── test_data_pipeline.py └── test_env.py ├── train.py └── trained_models ├── baseline └── model.pt ├── gnn_k2 ├── config.yaml └── model.pt ├── gnn_k3 ├── config.yaml └── model.pt └── gnn_msg_k3 ├── config.yaml └── model.pt /.github/workflows/ci.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/.github/workflows/ci.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/README.md -------------------------------------------------------------------------------- /benchmarks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /benchmarks/benchmark_performance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/benchmarks/benchmark_performance.py -------------------------------------------------------------------------------- /benchmarks/benchmark_scaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/benchmarks/benchmark_scaling.py -------------------------------------------------------------------------------- /cbs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cbs/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/cbs/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /cbs/__pycache__/a_star.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/cbs/__pycache__/a_star.cpython-39.pyc -------------------------------------------------------------------------------- /cbs/a_star.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/cbs/a_star.py -------------------------------------------------------------------------------- /cbs/cbs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/cbs/cbs.py -------------------------------------------------------------------------------- /cbs/examples/input.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/cbs/examples/input.yaml -------------------------------------------------------------------------------- /cbs/examples/output.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/cbs/examples/output.yaml -------------------------------------------------------------------------------- /cbs/examples/test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/cbs/examples/test.yaml -------------------------------------------------------------------------------- /cbs/examples/test_2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/cbs/examples/test_2.yaml -------------------------------------------------------------------------------- /cbs/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/cbs/visualize.py -------------------------------------------------------------------------------- /configs/config_baseline.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/configs/config_baseline.yaml -------------------------------------------------------------------------------- /configs/config_gnn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/configs/config_gnn.yaml -------------------------------------------------------------------------------- /data_generation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data_generation/dataset_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/data_generation/dataset_gen.py -------------------------------------------------------------------------------- /data_generation/main_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/data_generation/main_data.py -------------------------------------------------------------------------------- /data_generation/record.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/data_generation/record.py -------------------------------------------------------------------------------- /data_generation/trajectory_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/data_generation/trajectory_parser.py -------------------------------------------------------------------------------- /data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/data_loader.py -------------------------------------------------------------------------------- /demos/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/demos/README.md -------------------------------------------------------------------------------- /demos/cbs_env_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/demos/cbs_env_demo.py -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/evaluate.py -------------------------------------------------------------------------------- /example.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/example.gif -------------------------------------------------------------------------------- /example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/example.py -------------------------------------------------------------------------------- /grid/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /grid/controller_grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/grid/controller_grid.py -------------------------------------------------------------------------------- /grid/env_graph_grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/grid/env_graph_grid.py -------------------------------------------------------------------------------- /grid/env_graph_gridv1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/grid/env_graph_gridv1.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/framework_baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/models/framework_baseline.py -------------------------------------------------------------------------------- /models/framework_gnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/models/framework_gnn.py -------------------------------------------------------------------------------- /models/framework_gnn_message.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/models/framework_gnn_message.py -------------------------------------------------------------------------------- /models/networks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/networks/gnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/models/networks/gnn.py -------------------------------------------------------------------------------- /models/networks/utils_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/models/networks/utils_weights.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/requirements.txt -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_cbs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/tests/test_cbs.py -------------------------------------------------------------------------------- /tests/test_cbs_performance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/tests/test_cbs_performance.py -------------------------------------------------------------------------------- /tests/test_data_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/tests/test_data_pipeline.py -------------------------------------------------------------------------------- /tests/test_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/tests/test_env.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/train.py -------------------------------------------------------------------------------- /trained_models/baseline/model.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/trained_models/baseline/model.pt -------------------------------------------------------------------------------- /trained_models/gnn_k2/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/trained_models/gnn_k2/config.yaml -------------------------------------------------------------------------------- /trained_models/gnn_k2/model.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/trained_models/gnn_k2/model.pt -------------------------------------------------------------------------------- /trained_models/gnn_k3/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/trained_models/gnn_k3/config.yaml -------------------------------------------------------------------------------- /trained_models/gnn_k3/model.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/trained_models/gnn_k3/model.pt -------------------------------------------------------------------------------- /trained_models/gnn_msg_k3/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/trained_models/gnn_msg_k3/config.yaml -------------------------------------------------------------------------------- /trained_models/gnn_msg_k3/model.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RetamalVictor/MAPF-GNN/HEAD/trained_models/gnn_msg_k3/model.pt --------------------------------------------------------------------------------