├── .gitignore ├── LICENSE ├── README.md ├── configs ├── mlflow_runtime.py ├── shedules │ ├── detr.py │ └── detr3d.py └── submission │ ├── frozen_1 │ ├── base.py │ ├── query_center_proj_no_value_proj.py │ ├── query_center_proj_no_value_proj_shared.py │ ├── query_center_proj_no_value_proj_shared_cbgs.py │ ├── query_center_proj_no_value_proj_shared_cbgs_vovnet_trainval.py │ ├── query_center_proj_no_value_proj_shared_no_pretrain.py │ ├── query_center_proj_value_proj.py │ ├── query_center_proj_value_proj_shared.py │ ├── query_center_proj_value_proj_shared_lr8bs.py │ └── query_proj_value_proj.py │ └── frozen_4 │ ├── base.py │ ├── no_query_proj.py │ ├── query_center_proj_no_value_proj.py │ ├── query_center_proj_no_value_proj_shared.py │ ├── query_center_proj_value_proj.py │ ├── query_proj_no_value_proj.py │ ├── query_proj_value_proj.py │ ├── query_proj_value_proj_1200_queries.py │ ├── query_proj_value_proj_600_queries.py │ ├── query_proj_value_proj_dec_10.py │ ├── query_proj_value_proj_dec_10_shared.py │ ├── query_proj_value_proj_dec_2.py │ ├── query_proj_value_proj_glob_softmax.py │ ├── query_proj_value_proj_no_pretrain.py │ └── query_proj_value_proj_shared.py ├── docker ├── Dockerfile ├── build.sh ├── in_docker_setup.sh └── run_command.sh ├── getting_started.md ├── img ├── attention.png ├── overview.png └── qualitative_results.png ├── license_infos.md ├── setup.py ├── spatial_detr ├── __init__.py ├── models │ ├── __init__.py │ └── utils │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── positional_encoding.py │ │ └── spatial_detr_transformer.py ├── tests │ └── test_create_configs.py └── thirdparty │ ├── __init__.py │ ├── detr3d │ ├── LICENSE │ ├── __init__.py │ └── util.py │ └── torch │ ├── LICENSE │ ├── __init__.py │ ├── custom_multi_head_attention.py │ └── functional.py └── training_logs ├── frozen_1 ├── query_center_proj_no_value_proj_shared.log └── query_center_proj_no_value_proj_shared_cbgs_vovnet_trainval.log └── frozen_4 ├── query_proj_value_proj.log └── query_proj_value_proj_0_17.log /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/README.md -------------------------------------------------------------------------------- /configs/mlflow_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/mlflow_runtime.py -------------------------------------------------------------------------------- /configs/shedules/detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/shedules/detr.py -------------------------------------------------------------------------------- /configs/shedules/detr3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/shedules/detr3d.py -------------------------------------------------------------------------------- /configs/submission/frozen_1/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_1/base.py -------------------------------------------------------------------------------- /configs/submission/frozen_1/query_center_proj_no_value_proj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_1/query_center_proj_no_value_proj.py -------------------------------------------------------------------------------- /configs/submission/frozen_1/query_center_proj_no_value_proj_shared.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_1/query_center_proj_no_value_proj_shared.py -------------------------------------------------------------------------------- /configs/submission/frozen_1/query_center_proj_no_value_proj_shared_cbgs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_1/query_center_proj_no_value_proj_shared_cbgs.py -------------------------------------------------------------------------------- /configs/submission/frozen_1/query_center_proj_no_value_proj_shared_cbgs_vovnet_trainval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_1/query_center_proj_no_value_proj_shared_cbgs_vovnet_trainval.py -------------------------------------------------------------------------------- /configs/submission/frozen_1/query_center_proj_no_value_proj_shared_no_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_1/query_center_proj_no_value_proj_shared_no_pretrain.py -------------------------------------------------------------------------------- /configs/submission/frozen_1/query_center_proj_value_proj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_1/query_center_proj_value_proj.py -------------------------------------------------------------------------------- /configs/submission/frozen_1/query_center_proj_value_proj_shared.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_1/query_center_proj_value_proj_shared.py -------------------------------------------------------------------------------- /configs/submission/frozen_1/query_center_proj_value_proj_shared_lr8bs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_1/query_center_proj_value_proj_shared_lr8bs.py -------------------------------------------------------------------------------- /configs/submission/frozen_1/query_proj_value_proj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_1/query_proj_value_proj.py -------------------------------------------------------------------------------- /configs/submission/frozen_4/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_4/base.py -------------------------------------------------------------------------------- /configs/submission/frozen_4/no_query_proj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_4/no_query_proj.py -------------------------------------------------------------------------------- /configs/submission/frozen_4/query_center_proj_no_value_proj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_4/query_center_proj_no_value_proj.py -------------------------------------------------------------------------------- /configs/submission/frozen_4/query_center_proj_no_value_proj_shared.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_4/query_center_proj_no_value_proj_shared.py -------------------------------------------------------------------------------- /configs/submission/frozen_4/query_center_proj_value_proj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_4/query_center_proj_value_proj.py -------------------------------------------------------------------------------- /configs/submission/frozen_4/query_proj_no_value_proj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_4/query_proj_no_value_proj.py -------------------------------------------------------------------------------- /configs/submission/frozen_4/query_proj_value_proj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_4/query_proj_value_proj.py -------------------------------------------------------------------------------- /configs/submission/frozen_4/query_proj_value_proj_1200_queries.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_4/query_proj_value_proj_1200_queries.py -------------------------------------------------------------------------------- /configs/submission/frozen_4/query_proj_value_proj_600_queries.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_4/query_proj_value_proj_600_queries.py -------------------------------------------------------------------------------- /configs/submission/frozen_4/query_proj_value_proj_dec_10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_4/query_proj_value_proj_dec_10.py -------------------------------------------------------------------------------- /configs/submission/frozen_4/query_proj_value_proj_dec_10_shared.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_4/query_proj_value_proj_dec_10_shared.py -------------------------------------------------------------------------------- /configs/submission/frozen_4/query_proj_value_proj_dec_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_4/query_proj_value_proj_dec_2.py -------------------------------------------------------------------------------- /configs/submission/frozen_4/query_proj_value_proj_glob_softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_4/query_proj_value_proj_glob_softmax.py -------------------------------------------------------------------------------- /configs/submission/frozen_4/query_proj_value_proj_no_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_4/query_proj_value_proj_no_pretrain.py -------------------------------------------------------------------------------- /configs/submission/frozen_4/query_proj_value_proj_shared.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/configs/submission/frozen_4/query_proj_value_proj_shared.py -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /docker/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/docker/build.sh -------------------------------------------------------------------------------- /docker/in_docker_setup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/docker/in_docker_setup.sh -------------------------------------------------------------------------------- /docker/run_command.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/docker/run_command.sh -------------------------------------------------------------------------------- /getting_started.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/getting_started.md -------------------------------------------------------------------------------- /img/attention.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/img/attention.png -------------------------------------------------------------------------------- /img/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/img/overview.png -------------------------------------------------------------------------------- /img/qualitative_results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/img/qualitative_results.png -------------------------------------------------------------------------------- /license_infos.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/license_infos.md -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/setup.py -------------------------------------------------------------------------------- /spatial_detr/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /spatial_detr/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /spatial_detr/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/spatial_detr/models/utils/__init__.py -------------------------------------------------------------------------------- /spatial_detr/models/utils/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/spatial_detr/models/utils/attention.py -------------------------------------------------------------------------------- /spatial_detr/models/utils/positional_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/spatial_detr/models/utils/positional_encoding.py -------------------------------------------------------------------------------- /spatial_detr/models/utils/spatial_detr_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/spatial_detr/models/utils/spatial_detr_transformer.py -------------------------------------------------------------------------------- /spatial_detr/tests/test_create_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/spatial_detr/tests/test_create_configs.py -------------------------------------------------------------------------------- /spatial_detr/thirdparty/__init__.py: -------------------------------------------------------------------------------- 1 | # init for thirdparty libs -------------------------------------------------------------------------------- /spatial_detr/thirdparty/detr3d/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/spatial_detr/thirdparty/detr3d/LICENSE -------------------------------------------------------------------------------- /spatial_detr/thirdparty/detr3d/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /spatial_detr/thirdparty/detr3d/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/spatial_detr/thirdparty/detr3d/util.py -------------------------------------------------------------------------------- /spatial_detr/thirdparty/torch/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/spatial_detr/thirdparty/torch/LICENSE -------------------------------------------------------------------------------- /spatial_detr/thirdparty/torch/__init__.py: -------------------------------------------------------------------------------- 1 | # init for modification on torch -------------------------------------------------------------------------------- /spatial_detr/thirdparty/torch/custom_multi_head_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/spatial_detr/thirdparty/torch/custom_multi_head_attention.py -------------------------------------------------------------------------------- /spatial_detr/thirdparty/torch/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/spatial_detr/thirdparty/torch/functional.py -------------------------------------------------------------------------------- /training_logs/frozen_1/query_center_proj_no_value_proj_shared.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/training_logs/frozen_1/query_center_proj_no_value_proj_shared.log -------------------------------------------------------------------------------- /training_logs/frozen_1/query_center_proj_no_value_proj_shared_cbgs_vovnet_trainval.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/training_logs/frozen_1/query_center_proj_no_value_proj_shared_cbgs_vovnet_trainval.log -------------------------------------------------------------------------------- /training_logs/frozen_4/query_proj_value_proj.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/training_logs/frozen_4/query_proj_value_proj.log -------------------------------------------------------------------------------- /training_logs/frozen_4/query_proj_value_proj_0_17.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cgtuebingen/SpatialDETR/HEAD/training_logs/frozen_4/query_proj_value_proj_0_17.log --------------------------------------------------------------------------------