├── .gitignore ├── README.md ├── checkpoint.txt ├── config ├── DQ_5scale.py └── coco_transformer.py ├── datasets ├── __init__.py ├── coco.py ├── coco_eval.py ├── coco_panoptic.py ├── data_util.py ├── dataset.py ├── panoptic_eval.py ├── random_crop.py ├── sltransform.py └── transforms.py ├── engine.py ├── eval_best305.txt ├── figure └── model_final_V4.pdf ├── install.sh ├── logs └── DQ_eval │ ├── config_args_all.json │ ├── config_args_raw.json │ ├── config_cfg.py │ ├── info.txt │ └── log.txt ├── main_aitod.py ├── models ├── __init__.py ├── dqdetr │ ├── __init__.py │ ├── attention.py │ ├── backbone.py │ ├── ccm.py │ ├── cgfe.py │ ├── deformable_transformer.py │ ├── dn_components.py │ ├── dqdetr.py │ ├── matcher.py │ ├── ops │ │ ├── functions │ │ │ ├── __init__.py │ │ │ └── ms_deform_attn_func.py │ │ ├── make.sh │ │ ├── modules │ │ │ ├── __init__.py │ │ │ └── ms_deform_attn.py │ │ ├── setup.py │ │ ├── src │ │ │ ├── cpu │ │ │ │ ├── ms_deform_attn_cpu.cpp │ │ │ │ └── ms_deform_attn_cpu.h │ │ │ ├── cuda │ │ │ │ ├── ms_deform_attn_cuda.cu │ │ │ │ ├── ms_deform_attn_cuda.h │ │ │ │ └── ms_deform_im2col_cuda.cuh │ │ │ ├── ms_deform_attn.h │ │ │ └── vision.cpp │ │ └── test.py │ ├── position_encoding.py │ ├── segmentation.py │ ├── transformer_deformable.py │ └── utils.py └── registry.py ├── requirements.txt ├── scripts ├── DQ.sh └── DQ_eval.sh ├── tools └── benchmark.py └── util ├── __init__.py ├── box_loss.py ├── box_ops.py ├── coco_id2name.json ├── get_param_dicts.py ├── logger.py ├── misc.py ├── plot_utils.py ├── slconfig.py ├── slio.py ├── static_data_path.py ├── time_counter.py ├── utils.py ├── vis_utils.py └── visualizer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/README.md -------------------------------------------------------------------------------- /checkpoint.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/checkpoint.txt -------------------------------------------------------------------------------- /config/DQ_5scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/config/DQ_5scale.py -------------------------------------------------------------------------------- /config/coco_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/config/coco_transformer.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/datasets/coco.py -------------------------------------------------------------------------------- /datasets/coco_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/datasets/coco_eval.py -------------------------------------------------------------------------------- /datasets/coco_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/datasets/coco_panoptic.py -------------------------------------------------------------------------------- /datasets/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/datasets/data_util.py -------------------------------------------------------------------------------- /datasets/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/datasets/dataset.py -------------------------------------------------------------------------------- /datasets/panoptic_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/datasets/panoptic_eval.py -------------------------------------------------------------------------------- /datasets/random_crop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/datasets/random_crop.py -------------------------------------------------------------------------------- /datasets/sltransform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/datasets/sltransform.py -------------------------------------------------------------------------------- /datasets/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/datasets/transforms.py -------------------------------------------------------------------------------- /engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/engine.py -------------------------------------------------------------------------------- /eval_best305.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/eval_best305.txt -------------------------------------------------------------------------------- /figure/model_final_V4.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/figure/model_final_V4.pdf -------------------------------------------------------------------------------- /install.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/install.sh -------------------------------------------------------------------------------- /logs/DQ_eval/config_args_all.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/logs/DQ_eval/config_args_all.json -------------------------------------------------------------------------------- /logs/DQ_eval/config_args_raw.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/logs/DQ_eval/config_args_raw.json -------------------------------------------------------------------------------- /logs/DQ_eval/config_cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/logs/DQ_eval/config_cfg.py -------------------------------------------------------------------------------- /logs/DQ_eval/info.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/logs/DQ_eval/info.txt -------------------------------------------------------------------------------- /logs/DQ_eval/log.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/logs/DQ_eval/log.txt -------------------------------------------------------------------------------- /main_aitod.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/main_aitod.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/dqdetr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/__init__.py -------------------------------------------------------------------------------- /models/dqdetr/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/attention.py -------------------------------------------------------------------------------- /models/dqdetr/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/backbone.py -------------------------------------------------------------------------------- /models/dqdetr/ccm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/ccm.py -------------------------------------------------------------------------------- /models/dqdetr/cgfe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/cgfe.py -------------------------------------------------------------------------------- /models/dqdetr/deformable_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/deformable_transformer.py -------------------------------------------------------------------------------- /models/dqdetr/dn_components.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/dn_components.py -------------------------------------------------------------------------------- /models/dqdetr/dqdetr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/dqdetr.py -------------------------------------------------------------------------------- /models/dqdetr/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/matcher.py -------------------------------------------------------------------------------- /models/dqdetr/ops/functions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/ops/functions/__init__.py -------------------------------------------------------------------------------- /models/dqdetr/ops/functions/ms_deform_attn_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/ops/functions/ms_deform_attn_func.py -------------------------------------------------------------------------------- /models/dqdetr/ops/make.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/ops/make.sh -------------------------------------------------------------------------------- /models/dqdetr/ops/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/ops/modules/__init__.py -------------------------------------------------------------------------------- /models/dqdetr/ops/modules/ms_deform_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/ops/modules/ms_deform_attn.py -------------------------------------------------------------------------------- /models/dqdetr/ops/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/ops/setup.py -------------------------------------------------------------------------------- /models/dqdetr/ops/src/cpu/ms_deform_attn_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/ops/src/cpu/ms_deform_attn_cpu.cpp -------------------------------------------------------------------------------- /models/dqdetr/ops/src/cpu/ms_deform_attn_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/ops/src/cpu/ms_deform_attn_cpu.h -------------------------------------------------------------------------------- /models/dqdetr/ops/src/cuda/ms_deform_attn_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/ops/src/cuda/ms_deform_attn_cuda.cu -------------------------------------------------------------------------------- /models/dqdetr/ops/src/cuda/ms_deform_attn_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/ops/src/cuda/ms_deform_attn_cuda.h -------------------------------------------------------------------------------- /models/dqdetr/ops/src/cuda/ms_deform_im2col_cuda.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/ops/src/cuda/ms_deform_im2col_cuda.cuh -------------------------------------------------------------------------------- /models/dqdetr/ops/src/ms_deform_attn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/ops/src/ms_deform_attn.h -------------------------------------------------------------------------------- /models/dqdetr/ops/src/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/ops/src/vision.cpp -------------------------------------------------------------------------------- /models/dqdetr/ops/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/ops/test.py -------------------------------------------------------------------------------- /models/dqdetr/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/position_encoding.py -------------------------------------------------------------------------------- /models/dqdetr/segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/segmentation.py -------------------------------------------------------------------------------- /models/dqdetr/transformer_deformable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/transformer_deformable.py -------------------------------------------------------------------------------- /models/dqdetr/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/dqdetr/utils.py -------------------------------------------------------------------------------- /models/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/models/registry.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/DQ.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/scripts/DQ.sh -------------------------------------------------------------------------------- /scripts/DQ_eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/scripts/DQ_eval.sh -------------------------------------------------------------------------------- /tools/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/tools/benchmark.py -------------------------------------------------------------------------------- /util/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved 2 | -------------------------------------------------------------------------------- /util/box_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/util/box_loss.py -------------------------------------------------------------------------------- /util/box_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/util/box_ops.py -------------------------------------------------------------------------------- /util/coco_id2name.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/util/coco_id2name.json -------------------------------------------------------------------------------- /util/get_param_dicts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/util/get_param_dicts.py -------------------------------------------------------------------------------- /util/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/util/logger.py -------------------------------------------------------------------------------- /util/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/util/misc.py -------------------------------------------------------------------------------- /util/plot_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/util/plot_utils.py -------------------------------------------------------------------------------- /util/slconfig.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/util/slconfig.py -------------------------------------------------------------------------------- /util/slio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/util/slio.py -------------------------------------------------------------------------------- /util/static_data_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/util/static_data_path.py -------------------------------------------------------------------------------- /util/time_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/util/time_counter.py -------------------------------------------------------------------------------- /util/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/util/utils.py -------------------------------------------------------------------------------- /util/vis_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/util/vis_utils.py -------------------------------------------------------------------------------- /util/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hoiliu-0801/DQ-DETR/HEAD/util/visualizer.py --------------------------------------------------------------------------------