├── .clang-format ├── .flake8 ├── .gitignore ├── .gitmodules ├── .readthedocs.yaml ├── LICENSE ├── README.md ├── assets ├── detr_arch.png ├── detrex_logo.png └── logo_2.png ├── changlog.md ├── configs ├── common │ ├── coco_schedule.py │ ├── common_schedule.py │ ├── data │ │ ├── coco.py │ │ ├── coco_detr.py │ │ ├── constants.py │ │ └── custom.py │ ├── optim.py │ └── train.py └── hydra │ ├── slurm │ └── research.yaml │ └── train_args.yaml ├── demo ├── README.md ├── __init__.py ├── demo.py ├── mot_demo.py ├── mot_predictors.py └── predictors.py ├── detrex ├── __init__.py ├── checkpoint │ ├── __init__.py │ ├── c2_model_loading.py │ └── detection_checkpoint.py ├── config │ ├── __init__.py │ └── config.py ├── data │ ├── __init__.py │ ├── dataset_mappers │ │ ├── __init__.py │ │ ├── coco_instance_new_baseline_dataset_mapper.py │ │ ├── coco_panoptic_new_baseline_dataset_mapper.py │ │ ├── mask_former_instance_dataset_mapper.py │ │ ├── mask_former_panoptic_dataset_mapper.py │ │ └── mask_former_semantic_dataset_mapper.py │ ├── datasets │ │ ├── __init__.py │ │ ├── register_ade20k_full.py │ │ ├── register_ade20k_instance.py │ │ ├── register_ade20k_panoptic.py │ │ ├── register_coco_panoptic_annos_semseg.py │ │ ├── register_coco_stuff_10k.py │ │ ├── register_mapillary_vistas.py │ │ └── register_mapillary_vistas_panoptic.py │ ├── detr_dataset_mapper.py │ └── transforms │ │ ├── __init__.py │ │ └── color_augmentation.py ├── layers │ ├── __init__.py │ ├── attention.py │ ├── box_ops.py │ ├── conv.py │ ├── csrc │ │ ├── DCNv3 │ │ │ ├── dcnv3.h │ │ │ ├── dcnv3_cpu.cpp │ │ │ ├── dcnv3_cpu.h │ │ │ ├── dcnv3_cuda.cu │ │ │ ├── dcnv3_cuda.h │ │ │ └── dcnv3_im2col_cuda.cuh │ │ ├── MsDeformAttn │ │ │ ├── ms_deform_attn.h │ │ │ ├── ms_deform_attn_cpu.cpp │ │ │ ├── ms_deform_attn_cpu.h │ │ │ ├── ms_deform_attn_cuda.cu │ │ │ ├── ms_deform_attn_cuda.h │ │ │ └── ms_deform_im2col_cuda.cuh │ │ ├── cuda_version.cu │ │ └── vision.cpp │ ├── dcn_v3.py │ ├── denoising.py │ ├── layer_norm.py │ ├── mlp.py │ ├── multi_scale_deform_attn.py │ ├── position_embedding.py │ ├── shape_spec.py │ └── transformer.py ├── modeling │ ├── __init__.py │ ├── backbone │ │ ├── __init__.py │ │ ├── convnext.py │ │ ├── eva.py │ │ ├── eva_02.py │ │ ├── eva_02_utils.py │ │ ├── focalnet.py │ │ ├── internimage.py │ │ ├── resnet.py │ │ ├── timm_backbone.py │ │ ├── torchvision_backbone.py │ │ ├── torchvision_resnet.py │ │ └── utils.py │ ├── criterion │ │ ├── __init__.py │ │ ├── base_criterion.py │ │ └── criterion.py │ ├── ema.py │ ├── losses │ │ ├── __init__.py │ │ ├── cross_entropy_loss.py │ │ ├── dice_loss.py │ │ ├── focal_loss.py │ │ ├── giou_loss.py │ │ ├── smooth_l1_loss.py │ │ └── utils.py │ ├── matcher │ │ ├── __init__.py │ │ ├── match_cost.py │ │ ├── matcher.py │ │ └── modified_matcher.py │ └── neck │ │ ├── __init__.py │ │ └── channel_mapper.py └── utils │ ├── __init__.py │ ├── dist.py │ ├── events.py │ └── misc.py ├── dev ├── linter.sh └── run_unittest.sh ├── docs ├── Makefile ├── README.md ├── requirements.txt └── source │ ├── _static │ └── css │ │ └── line_space.css │ ├── _templates │ ├── .gitkeep │ └── line_space.html │ ├── changelog.md │ ├── conf.py │ ├── index.rst │ ├── modules │ ├── detrex.config.rst │ ├── detrex.data.rst │ ├── detrex.layers.rst │ ├── detrex.modeling.rst │ ├── detrex.utils.rst │ └── index.rst │ └── tutorials │ ├── Config_System.md │ ├── Converters.md │ ├── Customize_Training.md │ ├── Download_Pretrained_Weights.md │ ├── FAQs.md │ ├── Getting_Started.md │ ├── Installation.md │ ├── Model_Zoo.md │ ├── Tools.md │ ├── Using_Pretrained_Backbone.md │ ├── assets │ ├── annotation_demo.jpg │ ├── cosine_lr_scheduler.png │ ├── demo_output.jpg │ ├── dino_prediction_demo.jpg │ ├── exponential_lr_scheduler.png │ ├── linear_lr_scheduler.png │ ├── multi_step_example.png │ ├── multi_step_lr_scheduler.png │ ├── step_lr_scheduler.png │ └── step_lr_with_fixed_gamma.png │ └── index.rst ├── projects ├── README.md ├── align_detr │ ├── README.md │ ├── configs │ │ ├── aligndetr_k=2_r50_4scale_12ep.py │ │ ├── aligndetr_k=2_r50_4scale_24ep.py │ │ ├── aligndetr_k=2_r50_4scale_36ep.py │ │ └── models │ │ │ └── aligndetr_r50.py │ └── modeling │ │ ├── __init__.py │ │ ├── aligndetr.py │ │ ├── criterions │ │ ├── __init__.py │ │ ├── aligndetr_dn_criterion.py │ │ ├── base_criterion.py │ │ ├── many_to_one_criterion.py │ │ └── two_stage_criterion.py │ │ ├── losses │ │ ├── __init__.py │ │ └── losses.py │ │ ├── matchers │ │ ├── __init__.py │ │ └── mixed_matcher.py │ │ └── transformer.py ├── anchor_detr │ ├── README.md │ ├── assets │ │ └── anchor_detr_arch.png │ ├── configs │ │ ├── anchor_detr_r101_50ep.py │ │ ├── anchor_detr_r101_dc5_50ep.py │ │ ├── anchor_detr_r50_50ep.py │ │ ├── anchor_detr_r50_dc5_50ep.py │ │ └── models │ │ │ └── anchor_detr_r50.py │ └── modeling │ │ ├── __init__.py │ │ ├── anchor_detr.py │ │ ├── anchor_detr_transformer.py │ │ ├── row_column_decoupled_attention.py │ │ └── utils.py ├── co_mot │ ├── README.md │ ├── configs │ │ ├── common │ │ │ ├── dancetrack_schedule.py │ │ │ └── data │ │ │ │ └── dancetrack_mot.py │ │ ├── mot_r50.py │ │ └── mot_r50_4scale_10ep.py │ ├── data │ │ ├── __init__.py │ │ ├── datasets │ │ │ ├── __init__.py │ │ │ └── register_dancetrack_mot.py │ │ ├── mot_build.py │ │ ├── mot_dataset_mapper.py │ │ └── transforms │ │ │ ├── __init__.py │ │ │ └── mot_transforms.py │ ├── evaluation │ │ ├── __init__.py │ │ └── dancetrack_evaluation.py │ ├── modeling │ │ ├── __init__.py │ │ ├── matcher.py │ │ ├── mot.py │ │ ├── mot_transformer.py │ │ └── qim.py │ ├── train_net.py │ └── util │ │ ├── __init__.py │ │ ├── checkpoint.py │ │ └── misc.py ├── conditional_detr │ ├── README.md │ ├── assets │ │ └── attention-maps.png │ ├── configs │ │ ├── conditional_detr_r101_50ep.py │ │ ├── conditional_detr_r101_dc5_50ep.py │ │ ├── conditional_detr_r50_50ep.py │ │ ├── conditional_detr_r50_dc5_50ep.py │ │ └── models │ │ │ ├── conditional_detr_r50.py │ │ │ └── conditional_detr_r50_dc5.py │ ├── converter.py │ └── modeling │ │ ├── __init__.py │ │ ├── conditional_detr.py │ │ └── conditional_transformer.py ├── dab_deformable_detr │ ├── README.md │ ├── assets │ │ └── dab_detr_overall.png │ ├── configs │ │ ├── dab_deformable_detr_r50_50ep.py │ │ ├── dab_deformable_detr_r50_two_stage_50ep.py │ │ └── models │ │ │ └── dab_deformable_detr_r50.py │ └── modeling │ │ ├── __init__.py │ │ ├── dab_deformable_detr.py │ │ ├── dab_deformable_transformer.py │ │ └── two_stage_criterion.py ├── dab_detr │ ├── README.md │ ├── assets │ │ ├── dab_detr_details.png │ │ └── dab_detr_overall.png │ ├── configs │ │ ├── dab_detr_r101_50ep.py │ │ ├── dab_detr_r101_dc5_50ep.py │ │ ├── dab_detr_r50_3patterns_50ep.py │ │ ├── dab_detr_r50_50ep.py │ │ ├── dab_detr_r50_dc5_3patterns_50ep.py │ │ ├── dab_detr_r50_dc5_50ep.py │ │ ├── dab_detr_swin_b_in21k_50ep.py │ │ ├── dab_detr_swin_t_in1k_50ep.py │ │ └── models │ │ │ ├── dab_detr_r50.py │ │ │ ├── dab_detr_r50_3patterns.py │ │ │ ├── dab_detr_r50_dc5.py │ │ │ ├── dab_detr_swin_base.py │ │ │ └── dab_detr_swin_tiny.py │ └── modeling │ │ ├── __init__.py │ │ ├── dab_detr.py │ │ └── dab_transformer.py ├── deformable_detr │ ├── README.md │ ├── assets │ │ └── deformable_detr.png │ ├── configs │ │ ├── deformable_detr_r50_50ep.py │ │ ├── deformable_detr_r50_two_stage_50ep.py │ │ ├── deformable_detr_r50_with_box_refinement_50ep.py │ │ └── models │ │ │ └── deformable_detr_r50.py │ ├── convert_two_stage.py │ ├── converter.py │ ├── modeling │ │ ├── __init__.py │ │ ├── deformable_criterion.py │ │ ├── deformable_detr.py │ │ └── deformable_transformer.py │ └── train_net.py ├── deta │ ├── README.md │ ├── assets │ │ └── deta.png │ ├── configs │ │ ├── data │ │ │ └── coco_detr_larger.py │ │ ├── deta_r50_5scale_12ep.py │ │ ├── deta_r50_5scale_12ep_bs8.py │ │ ├── deta_r50_5scale_no_frozen_backbone.py │ │ ├── deta_swin_large_finetune_24ep.py │ │ ├── improved_deformable_detr_baseline_50ep.py │ │ ├── models │ │ │ ├── deta_r50.py │ │ │ └── deta_swin.py │ │ └── scheduler │ │ │ └── coco_scheduler.py │ ├── modeling │ │ ├── __init__.py │ │ ├── assigner.py │ │ ├── deformable_detr.py │ │ ├── deformable_transformer.py │ │ └── deta_criterion.py │ └── train_net.py ├── detr │ ├── README.md │ ├── assets │ │ └── DETR.png │ ├── configs │ │ ├── detr_r101_300ep.py │ │ ├── detr_r101_dc5_300ep.py │ │ ├── detr_r50_300ep.py │ │ ├── detr_r50_dc5_300ep.py │ │ └── models │ │ │ ├── detr_r50.py │ │ │ └── detr_r50_dc5.py │ ├── converter.py │ └── modeling │ │ ├── __init__.py │ │ ├── detr.py │ │ └── transformer.py ├── dino │ ├── README.md │ ├── assets │ │ └── dino_arch.png │ ├── configs │ │ ├── dino-convnext │ │ │ ├── dino_convnext_base_384_4scale_12ep.py │ │ │ ├── dino_convnext_large_384_4scale_12ep.py │ │ │ ├── dino_convnext_small_384_4scale_12ep.py │ │ │ └── dino_convnext_tiny_384_4scale_12ep.py │ │ ├── dino-eva-01 │ │ │ ├── dino_eva_01_1536_4scale_12ep.py │ │ │ └── dino_eva_01_4scale_12ep.py │ │ ├── dino-focal │ │ │ ├── dino_focal_base_lrf_fl3_4scale_12ep.py │ │ │ ├── dino_focal_small_lrf_fl3_4scale_12ep.py │ │ │ ├── dino_focal_tiny_lrf_fl3_4scale_12ep.py │ │ │ ├── dino_focalnet_large_lrf_384_4scale_12ep.py │ │ │ ├── dino_focalnet_large_lrf_384_4scale_36ep.py │ │ │ ├── dino_focalnet_large_lrf_384_fl4_4scale_12ep.py │ │ │ ├── dino_focalnet_large_lrf_384_fl4_5scale_12ep.py │ │ │ ├── dino_focalnet_large_lrf_384_fl4_5scale_36ep.py │ │ │ └── focalnet.py │ │ ├── dino-internimage │ │ │ ├── dino_internimage_base_4scale_12ep.py │ │ │ ├── dino_internimage_large_4scale_12ep.py │ │ │ ├── dino_internimage_small_4scale_12ep.py │ │ │ └── dino_internimage_tiny_4scale_12ep.py │ │ ├── dino-resnet │ │ │ ├── dino_r101_4scale_12ep.py │ │ │ ├── dino_r50_4scale_12ep.py │ │ │ ├── dino_r50_4scale_12ep_300dn.py │ │ │ ├── dino_r50_4scale_12ep_better_hyper.py │ │ │ ├── dino_r50_4scale_12ep_no_frozen.py │ │ │ ├── dino_r50_4scale_24ep.py │ │ │ └── dino_r50_5scale_12ep.py │ │ ├── dino-swin │ │ │ ├── dino_swin_base_384_4scale_12ep.py │ │ │ ├── dino_swin_large_224_4scale_12ep.py │ │ │ ├── dino_swin_large_384_4scale_12ep.py │ │ │ ├── dino_swin_large_384_4scale_36ep.py │ │ │ ├── dino_swin_large_384_5scale_12ep.py │ │ │ ├── dino_swin_large_384_5scale_36ep.py │ │ │ ├── dino_swin_small_224_4scale_12ep.py │ │ │ └── dino_swin_tiny_224_4scale_12ep.py │ │ ├── dino-vitdet │ │ │ ├── dino_vitdet_base_4scale_12ep.py │ │ │ ├── dino_vitdet_base_4scale_50ep.py │ │ │ ├── dino_vitdet_large_4scale_12ep.py │ │ │ └── dino_vitdet_large_4scale_50ep.py │ │ ├── models │ │ │ ├── dino_convnext.py │ │ │ ├── dino_eva_01.py │ │ │ ├── dino_focalnet.py │ │ │ ├── dino_internimage.py │ │ │ ├── dino_r50.py │ │ │ ├── dino_swin_base_384.py │ │ │ ├── dino_swin_large_224.py │ │ │ ├── dino_swin_large_384.py │ │ │ ├── dino_swin_small_224.py │ │ │ ├── dino_swin_tiny_224.py │ │ │ └── dino_vitdet.py │ │ ├── timm_example.py │ │ └── torchvision_example.py │ ├── modeling │ │ ├── __init__.py │ │ ├── dino.py │ │ ├── dino_transformer.py │ │ ├── dn_criterion.py │ │ └── two_stage_criterion.py │ └── train_net.py ├── dino_eva │ ├── README.md │ ├── assets │ │ └── dino_arch.png │ ├── configs │ │ ├── common │ │ │ ├── coco_loader_lsj.py │ │ │ ├── coco_loader_lsj_1024.py │ │ │ ├── coco_loader_lsj_1280.py │ │ │ └── coco_loader_lsj_1536.py │ │ ├── dino-eva-01 │ │ │ ├── dino_eva_01_1280_4scale_12ep.py │ │ │ └── dino_eva_01_1536_4scale_12ep.py │ │ ├── dino-eva-02 │ │ │ ├── dino_eva_02_vitdet_b_4attn_1024_lrd0p7_4scale_12ep.py │ │ │ ├── dino_eva_02_vitdet_b_6attn_win32_1536_lrd0p7_4scale_12ep.py │ │ │ ├── dino_eva_02_vitdet_l_4attn_1024_lrd0p8_4scale_12ep.py │ │ │ ├── dino_eva_02_vitdet_l_8attn_1536_lrd0p8_4scale_12ep.py │ │ │ └── dino_eva_02_vitdet_l_8attn_win32_1536_lrd0p8_4scale_12ep.py │ │ └── models │ │ │ ├── dino_eva_01.py │ │ │ ├── dino_eva_02.py │ │ │ └── dino_r50.py │ ├── modeling │ │ ├── __init__.py │ │ ├── dino.py │ │ ├── dino_transformer.py │ │ ├── dn_criterion.py │ │ └── two_stage_criterion.py │ └── train_net.py ├── dn_deformable_detr │ ├── README.md │ ├── assets │ │ └── dn_detr_arch.png │ ├── configs │ │ ├── dn_deformable_detr_r50_12ep.py │ │ ├── dn_deformable_detr_r50_50ep.py │ │ └── models │ │ │ └── dn_deformable_detr_r50.py │ ├── converter.py │ └── modeling │ │ ├── __init__.py │ │ ├── dn_criterion.py │ │ ├── dn_deformable_detr.py │ │ └── dn_deformable_transformer.py ├── dn_detr │ ├── README.md │ ├── assets │ │ └── dn_detr_arch.png │ ├── configs │ │ ├── dn_detr_r50_50ep.py │ │ ├── dn_detr_r50_dc5_50ep.py │ │ └── models │ │ │ ├── dn_detr_r50.py │ │ │ └── dn_detr_r50_dc5.py │ └── modeling │ │ ├── __init__.py │ │ ├── dn_criterion.py │ │ ├── dn_detr.py │ │ └── dn_transformers.py ├── focus_detr │ ├── README.md │ ├── configs │ │ ├── focus_detr_resnet │ │ │ ├── focus_detr_r101_4scale_12ep.py │ │ │ ├── focus_detr_r101_4scale_24ep.py │ │ │ ├── focus_detr_r101_4scale_36ep.py │ │ │ ├── focus_detr_r50_4scale_12ep.py │ │ │ ├── focus_detr_r50_4scale_24ep.py │ │ │ └── focus_detr_r50_4scale_36ep.py │ │ ├── focus_detr_swin │ │ │ ├── focus_detr_swin_base_224_4scale_36ep.py │ │ │ ├── focus_detr_swin_base_384_4scale_36ep.py │ │ │ ├── focus_detr_swin_large_384_4scale_36ep.py │ │ │ ├── focus_detr_swin_tiny_224_4scale_12ep.py │ │ │ ├── focus_detr_swin_tiny_224_4scale_22k_12ep.py │ │ │ ├── focus_detr_swin_tiny_224_4scale_22k_36ep.py │ │ │ ├── focus_detr_swin_tiny_224_4scale_24ep.py │ │ │ └── focus_detr_swin_tiny_224_4scale_36ep.py │ │ └── models │ │ │ ├── focus_detr_r50.py │ │ │ ├── focus_detr_swin_base_384.py │ │ │ ├── focus_detr_swin_large_224.py │ │ │ ├── focus_detr_swin_large_384.py │ │ │ ├── focus_detr_swin_small_224.py │ │ │ └── focus_detr_swin_tiny_224.py │ └── modeling │ │ ├── __init__.py │ │ ├── dn_criterion.py │ │ ├── focus_detr.py │ │ ├── focus_detr_transformer.py │ │ ├── foreground_supervision.py │ │ ├── transformer_layer.py │ │ └── two_stage_criterion.py ├── group_detr │ ├── README.md │ ├── assets │ │ └── group_detr_arch.png │ ├── configs │ │ ├── group_detr_r50_50ep.py │ │ └── models │ │ │ └── group_detr_r50.py │ └── modeling │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── group_criterion.py │ │ ├── group_detr.py │ │ ├── group_detr_transformer.py │ │ └── group_matcher.py ├── h_deformable_detr │ ├── README.md │ ├── assets │ │ └── h_detr_arch.png │ ├── configs │ │ ├── h_deformable_detr_r50_50ep.py │ │ ├── h_deformable_detr_r50_two_stage_12ep.py │ │ ├── h_deformable_detr_r50_two_stage_36ep.py │ │ ├── h_deformable_detr_swin_large_two_stage_12ep.py │ │ ├── h_deformable_detr_swin_large_two_stage_12ep_900queries.py │ │ ├── h_deformable_detr_swin_large_two_stage_36ep.py │ │ ├── h_deformable_detr_swin_large_two_stage_36ep_900queries.py │ │ ├── h_deformable_detr_swin_tiny_two_stage_12ep.py │ │ ├── h_deformable_detr_swin_tiny_two_stage_36ep.py │ │ └── models │ │ │ └── h_deformable_detr_r50.py │ ├── modeling │ │ ├── __init__.py │ │ ├── deformable_criterion.py │ │ ├── h_deformable_detr.py │ │ └── h_deformable_transformer.py │ └── train_net.py ├── maskdino │ ├── README.md │ ├── assets │ │ ├── dinosaur.png │ │ ├── framework.jpg │ │ ├── instance.png │ │ ├── panoptic.png │ │ ├── semantic.png │ │ └── sota.png │ ├── configs │ │ ├── data │ │ │ ├── ade20k_semantic_seg.py │ │ │ ├── coco_instance_seg.py │ │ │ └── coco_panoptic_seg.py │ │ ├── maskdino_r50_ade20k_semantic_seg_160k.py │ │ ├── maskdino_r50_coco_instance_seg_50ep.py │ │ ├── maskdino_r50_coco_panoptic_seg_50ep.py │ │ ├── maskdino_r50_instance_seg_50ep.py │ │ └── models │ │ │ └── maskdino_r50.py │ ├── data │ │ ├── __init__.py │ │ └── dataset_mappers │ │ │ ├── __init__.py │ │ │ └── coco_instance_lsj_aug_dataset_mapper.py │ ├── evaluation │ │ ├── __init__.py │ │ └── instance_evaluation.py │ ├── maskdino.py │ ├── modeling │ │ ├── __init__.py │ │ ├── backbone │ │ │ ├── __init__.py │ │ │ ├── focal.py │ │ │ └── swin.py │ │ ├── criterion.py │ │ ├── matcher.py │ │ ├── meta_arch │ │ │ ├── __init__.py │ │ │ └── maskdino_head.py │ │ ├── pixel_decoder │ │ │ ├── __init__.py │ │ │ ├── maskdino_encoder.py │ │ │ └── position_encoding.py │ │ └── transformer_decoder │ │ │ ├── __init__.py │ │ │ ├── dino_decoder.py │ │ │ ├── maskdino_decoder.py │ │ │ └── utils.py │ └── utils │ │ ├── __init__.py │ │ ├── box_ops.py │ │ ├── misc.py │ │ └── utils.py ├── pnp_detr │ ├── README.md │ ├── assets │ │ └── PnP-DETR.png │ ├── configs │ │ ├── models │ │ │ └── pnp_detr_r50.py │ │ ├── pnp_detr_r101_300ep.py │ │ └── pnp_detr_r50_300ep.py │ └── modeling │ │ ├── __init__.py │ │ ├── detr.py │ │ └── transformer.py ├── rank_detr │ ├── README.md │ ├── assets │ │ └── rank_detr_overview.png │ ├── configs │ │ ├── models │ │ │ └── rank_detr_r50.py │ │ ├── rank_detr_r50_50ep.py │ │ ├── rank_detr_r50_two_stage_12ep.py │ │ ├── rank_detr_r50_two_stage_36ep.py │ │ ├── rank_detr_swin_large_two_stage_12ep.py │ │ ├── rank_detr_swin_large_two_stage_36ep.py │ │ ├── rank_detr_swin_tiny_two_stage_12ep.py │ │ └── rank_detr_swin_tiny_two_stage_36ep.py │ ├── modeling │ │ ├── __init__.py │ │ ├── high_order_matcher.py │ │ ├── rank_detr.py │ │ ├── rank_transformer.py │ │ └── rankdetr_criterion.py │ └── train_net.py └── sqr_detr │ ├── README.md │ ├── assets │ └── sqr_detr_overall.png │ ├── configs │ ├── dab_detr_r50_50ep_sqr.py │ └── models │ │ └── dab_detr_r50_sqr.py │ └── modeling │ ├── __init__.py │ └── dab_transformer_sqr.py ├── requirements.txt ├── setup.cfg ├── setup.py ├── tests ├── test_cond_attn.py ├── test_ffn.py ├── test_losses.py ├── test_ms_deform_attn.py ├── test_position_embedding.py ├── test_torchvision_backbone.py ├── test_transformer.py └── utils │ ├── __init__.py │ ├── attention.py │ ├── losses.py │ ├── mlp.py │ ├── potision_embedding.py │ └── transformer.py └── tools ├── README.md ├── __init__.py ├── analyze_model.py ├── benchmark.py ├── hydra_train_net.py ├── train_net.py ├── visualize_data.py └── visualize_json_results.py /.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/.clang-format -------------------------------------------------------------------------------- /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/.flake8 -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/.gitmodules -------------------------------------------------------------------------------- /.readthedocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/.readthedocs.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/README.md -------------------------------------------------------------------------------- /assets/detr_arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/assets/detr_arch.png -------------------------------------------------------------------------------- /assets/detrex_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/assets/detrex_logo.png -------------------------------------------------------------------------------- /assets/logo_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/assets/logo_2.png -------------------------------------------------------------------------------- /changlog.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/changlog.md -------------------------------------------------------------------------------- /configs/common/coco_schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/configs/common/coco_schedule.py -------------------------------------------------------------------------------- /configs/common/common_schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/configs/common/common_schedule.py -------------------------------------------------------------------------------- /configs/common/data/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/configs/common/data/coco.py -------------------------------------------------------------------------------- /configs/common/data/coco_detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/configs/common/data/coco_detr.py -------------------------------------------------------------------------------- /configs/common/data/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/configs/common/data/constants.py -------------------------------------------------------------------------------- /configs/common/data/custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/configs/common/data/custom.py -------------------------------------------------------------------------------- /configs/common/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/configs/common/optim.py -------------------------------------------------------------------------------- /configs/common/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/configs/common/train.py -------------------------------------------------------------------------------- /configs/hydra/slurm/research.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/configs/hydra/slurm/research.yaml -------------------------------------------------------------------------------- /configs/hydra/train_args.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/configs/hydra/train_args.yaml -------------------------------------------------------------------------------- /demo/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/demo/README.md -------------------------------------------------------------------------------- /demo/__init__.py: -------------------------------------------------------------------------------- 1 | from .predictors import VisualizationDemo 2 | -------------------------------------------------------------------------------- /demo/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/demo/demo.py -------------------------------------------------------------------------------- /demo/mot_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/demo/mot_demo.py -------------------------------------------------------------------------------- /demo/mot_predictors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/demo/mot_predictors.py -------------------------------------------------------------------------------- /demo/predictors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/demo/predictors.py -------------------------------------------------------------------------------- /detrex/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/__init__.py -------------------------------------------------------------------------------- /detrex/checkpoint/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/checkpoint/__init__.py -------------------------------------------------------------------------------- /detrex/checkpoint/c2_model_loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/checkpoint/c2_model_loading.py -------------------------------------------------------------------------------- /detrex/checkpoint/detection_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/checkpoint/detection_checkpoint.py -------------------------------------------------------------------------------- /detrex/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/config/__init__.py -------------------------------------------------------------------------------- /detrex/config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/config/config.py -------------------------------------------------------------------------------- /detrex/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/data/__init__.py -------------------------------------------------------------------------------- /detrex/data/dataset_mappers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/data/dataset_mappers/__init__.py -------------------------------------------------------------------------------- /detrex/data/dataset_mappers/coco_instance_new_baseline_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/data/dataset_mappers/coco_instance_new_baseline_dataset_mapper.py -------------------------------------------------------------------------------- /detrex/data/dataset_mappers/coco_panoptic_new_baseline_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/data/dataset_mappers/coco_panoptic_new_baseline_dataset_mapper.py -------------------------------------------------------------------------------- /detrex/data/dataset_mappers/mask_former_instance_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/data/dataset_mappers/mask_former_instance_dataset_mapper.py -------------------------------------------------------------------------------- /detrex/data/dataset_mappers/mask_former_panoptic_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/data/dataset_mappers/mask_former_panoptic_dataset_mapper.py -------------------------------------------------------------------------------- /detrex/data/dataset_mappers/mask_former_semantic_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/data/dataset_mappers/mask_former_semantic_dataset_mapper.py -------------------------------------------------------------------------------- /detrex/data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/data/datasets/__init__.py -------------------------------------------------------------------------------- /detrex/data/datasets/register_ade20k_full.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/data/datasets/register_ade20k_full.py -------------------------------------------------------------------------------- /detrex/data/datasets/register_ade20k_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/data/datasets/register_ade20k_instance.py -------------------------------------------------------------------------------- /detrex/data/datasets/register_ade20k_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/data/datasets/register_ade20k_panoptic.py -------------------------------------------------------------------------------- /detrex/data/datasets/register_coco_panoptic_annos_semseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/data/datasets/register_coco_panoptic_annos_semseg.py -------------------------------------------------------------------------------- /detrex/data/datasets/register_coco_stuff_10k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/data/datasets/register_coco_stuff_10k.py -------------------------------------------------------------------------------- /detrex/data/datasets/register_mapillary_vistas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/data/datasets/register_mapillary_vistas.py -------------------------------------------------------------------------------- /detrex/data/datasets/register_mapillary_vistas_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/data/datasets/register_mapillary_vistas_panoptic.py -------------------------------------------------------------------------------- /detrex/data/detr_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/data/detr_dataset_mapper.py -------------------------------------------------------------------------------- /detrex/data/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/data/transforms/__init__.py -------------------------------------------------------------------------------- /detrex/data/transforms/color_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/data/transforms/color_augmentation.py -------------------------------------------------------------------------------- /detrex/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/__init__.py -------------------------------------------------------------------------------- /detrex/layers/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/attention.py -------------------------------------------------------------------------------- /detrex/layers/box_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/box_ops.py -------------------------------------------------------------------------------- /detrex/layers/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/conv.py -------------------------------------------------------------------------------- /detrex/layers/csrc/DCNv3/dcnv3.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/csrc/DCNv3/dcnv3.h -------------------------------------------------------------------------------- /detrex/layers/csrc/DCNv3/dcnv3_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/csrc/DCNv3/dcnv3_cpu.cpp -------------------------------------------------------------------------------- /detrex/layers/csrc/DCNv3/dcnv3_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/csrc/DCNv3/dcnv3_cpu.h -------------------------------------------------------------------------------- /detrex/layers/csrc/DCNv3/dcnv3_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/csrc/DCNv3/dcnv3_cuda.cu -------------------------------------------------------------------------------- /detrex/layers/csrc/DCNv3/dcnv3_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/csrc/DCNv3/dcnv3_cuda.h -------------------------------------------------------------------------------- /detrex/layers/csrc/DCNv3/dcnv3_im2col_cuda.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/csrc/DCNv3/dcnv3_im2col_cuda.cuh -------------------------------------------------------------------------------- /detrex/layers/csrc/MsDeformAttn/ms_deform_attn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/csrc/MsDeformAttn/ms_deform_attn.h -------------------------------------------------------------------------------- /detrex/layers/csrc/MsDeformAttn/ms_deform_attn_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/csrc/MsDeformAttn/ms_deform_attn_cpu.cpp -------------------------------------------------------------------------------- /detrex/layers/csrc/MsDeformAttn/ms_deform_attn_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/csrc/MsDeformAttn/ms_deform_attn_cpu.h -------------------------------------------------------------------------------- /detrex/layers/csrc/MsDeformAttn/ms_deform_attn_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/csrc/MsDeformAttn/ms_deform_attn_cuda.cu -------------------------------------------------------------------------------- /detrex/layers/csrc/MsDeformAttn/ms_deform_attn_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/csrc/MsDeformAttn/ms_deform_attn_cuda.h -------------------------------------------------------------------------------- /detrex/layers/csrc/MsDeformAttn/ms_deform_im2col_cuda.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/csrc/MsDeformAttn/ms_deform_im2col_cuda.cuh -------------------------------------------------------------------------------- /detrex/layers/csrc/cuda_version.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/csrc/cuda_version.cu -------------------------------------------------------------------------------- /detrex/layers/csrc/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/csrc/vision.cpp -------------------------------------------------------------------------------- /detrex/layers/dcn_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/dcn_v3.py -------------------------------------------------------------------------------- /detrex/layers/denoising.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/denoising.py -------------------------------------------------------------------------------- /detrex/layers/layer_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/layer_norm.py -------------------------------------------------------------------------------- /detrex/layers/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/mlp.py -------------------------------------------------------------------------------- /detrex/layers/multi_scale_deform_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/multi_scale_deform_attn.py -------------------------------------------------------------------------------- /detrex/layers/position_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/position_embedding.py -------------------------------------------------------------------------------- /detrex/layers/shape_spec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/shape_spec.py -------------------------------------------------------------------------------- /detrex/layers/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/layers/transformer.py -------------------------------------------------------------------------------- /detrex/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/__init__.py -------------------------------------------------------------------------------- /detrex/modeling/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/backbone/__init__.py -------------------------------------------------------------------------------- /detrex/modeling/backbone/convnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/backbone/convnext.py -------------------------------------------------------------------------------- /detrex/modeling/backbone/eva.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/backbone/eva.py -------------------------------------------------------------------------------- /detrex/modeling/backbone/eva_02.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/backbone/eva_02.py -------------------------------------------------------------------------------- /detrex/modeling/backbone/eva_02_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/backbone/eva_02_utils.py -------------------------------------------------------------------------------- /detrex/modeling/backbone/focalnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/backbone/focalnet.py -------------------------------------------------------------------------------- /detrex/modeling/backbone/internimage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/backbone/internimage.py -------------------------------------------------------------------------------- /detrex/modeling/backbone/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/backbone/resnet.py -------------------------------------------------------------------------------- /detrex/modeling/backbone/timm_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/backbone/timm_backbone.py -------------------------------------------------------------------------------- /detrex/modeling/backbone/torchvision_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/backbone/torchvision_backbone.py -------------------------------------------------------------------------------- /detrex/modeling/backbone/torchvision_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/backbone/torchvision_resnet.py -------------------------------------------------------------------------------- /detrex/modeling/backbone/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/backbone/utils.py -------------------------------------------------------------------------------- /detrex/modeling/criterion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/criterion/__init__.py -------------------------------------------------------------------------------- /detrex/modeling/criterion/base_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/criterion/base_criterion.py -------------------------------------------------------------------------------- /detrex/modeling/criterion/criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/criterion/criterion.py -------------------------------------------------------------------------------- /detrex/modeling/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/ema.py -------------------------------------------------------------------------------- /detrex/modeling/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/losses/__init__.py -------------------------------------------------------------------------------- /detrex/modeling/losses/cross_entropy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/losses/cross_entropy_loss.py -------------------------------------------------------------------------------- /detrex/modeling/losses/dice_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/losses/dice_loss.py -------------------------------------------------------------------------------- /detrex/modeling/losses/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/losses/focal_loss.py -------------------------------------------------------------------------------- /detrex/modeling/losses/giou_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/losses/giou_loss.py -------------------------------------------------------------------------------- /detrex/modeling/losses/smooth_l1_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/losses/smooth_l1_loss.py -------------------------------------------------------------------------------- /detrex/modeling/losses/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/losses/utils.py -------------------------------------------------------------------------------- /detrex/modeling/matcher/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/matcher/__init__.py -------------------------------------------------------------------------------- /detrex/modeling/matcher/match_cost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/matcher/match_cost.py -------------------------------------------------------------------------------- /detrex/modeling/matcher/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/matcher/matcher.py -------------------------------------------------------------------------------- /detrex/modeling/matcher/modified_matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/matcher/modified_matcher.py -------------------------------------------------------------------------------- /detrex/modeling/neck/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/neck/__init__.py -------------------------------------------------------------------------------- /detrex/modeling/neck/channel_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/modeling/neck/channel_mapper.py -------------------------------------------------------------------------------- /detrex/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/utils/__init__.py -------------------------------------------------------------------------------- /detrex/utils/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/utils/dist.py -------------------------------------------------------------------------------- /detrex/utils/events.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/utils/events.py -------------------------------------------------------------------------------- /detrex/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/detrex/utils/misc.py -------------------------------------------------------------------------------- /dev/linter.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/dev/linter.sh -------------------------------------------------------------------------------- /dev/run_unittest.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/dev/run_unittest.sh -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /docs/source/_static/css/line_space.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/_static/css/line_space.css -------------------------------------------------------------------------------- /docs/source/_templates/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docs/source/_templates/line_space.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/_templates/line_space.html -------------------------------------------------------------------------------- /docs/source/changelog.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/changelog.md -------------------------------------------------------------------------------- /docs/source/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/conf.py -------------------------------------------------------------------------------- /docs/source/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/index.rst -------------------------------------------------------------------------------- /docs/source/modules/detrex.config.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/modules/detrex.config.rst -------------------------------------------------------------------------------- /docs/source/modules/detrex.data.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/modules/detrex.data.rst -------------------------------------------------------------------------------- /docs/source/modules/detrex.layers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/modules/detrex.layers.rst -------------------------------------------------------------------------------- /docs/source/modules/detrex.modeling.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/modules/detrex.modeling.rst -------------------------------------------------------------------------------- /docs/source/modules/detrex.utils.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/modules/detrex.utils.rst -------------------------------------------------------------------------------- /docs/source/modules/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/modules/index.rst -------------------------------------------------------------------------------- /docs/source/tutorials/Config_System.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/tutorials/Config_System.md -------------------------------------------------------------------------------- /docs/source/tutorials/Converters.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/tutorials/Converters.md -------------------------------------------------------------------------------- /docs/source/tutorials/Customize_Training.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/tutorials/Customize_Training.md -------------------------------------------------------------------------------- /docs/source/tutorials/Download_Pretrained_Weights.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/tutorials/Download_Pretrained_Weights.md -------------------------------------------------------------------------------- /docs/source/tutorials/FAQs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/tutorials/FAQs.md -------------------------------------------------------------------------------- /docs/source/tutorials/Getting_Started.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/tutorials/Getting_Started.md -------------------------------------------------------------------------------- /docs/source/tutorials/Installation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/tutorials/Installation.md -------------------------------------------------------------------------------- /docs/source/tutorials/Model_Zoo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/tutorials/Model_Zoo.md -------------------------------------------------------------------------------- /docs/source/tutorials/Tools.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/tutorials/Tools.md -------------------------------------------------------------------------------- /docs/source/tutorials/Using_Pretrained_Backbone.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/tutorials/Using_Pretrained_Backbone.md -------------------------------------------------------------------------------- /docs/source/tutorials/assets/annotation_demo.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/tutorials/assets/annotation_demo.jpg -------------------------------------------------------------------------------- /docs/source/tutorials/assets/cosine_lr_scheduler.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/tutorials/assets/cosine_lr_scheduler.png -------------------------------------------------------------------------------- /docs/source/tutorials/assets/demo_output.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/tutorials/assets/demo_output.jpg -------------------------------------------------------------------------------- /docs/source/tutorials/assets/dino_prediction_demo.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/tutorials/assets/dino_prediction_demo.jpg -------------------------------------------------------------------------------- /docs/source/tutorials/assets/exponential_lr_scheduler.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/tutorials/assets/exponential_lr_scheduler.png -------------------------------------------------------------------------------- /docs/source/tutorials/assets/linear_lr_scheduler.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/tutorials/assets/linear_lr_scheduler.png -------------------------------------------------------------------------------- /docs/source/tutorials/assets/multi_step_example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/tutorials/assets/multi_step_example.png -------------------------------------------------------------------------------- /docs/source/tutorials/assets/multi_step_lr_scheduler.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/tutorials/assets/multi_step_lr_scheduler.png -------------------------------------------------------------------------------- /docs/source/tutorials/assets/step_lr_scheduler.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/tutorials/assets/step_lr_scheduler.png -------------------------------------------------------------------------------- /docs/source/tutorials/assets/step_lr_with_fixed_gamma.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/tutorials/assets/step_lr_with_fixed_gamma.png -------------------------------------------------------------------------------- /docs/source/tutorials/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/docs/source/tutorials/index.rst -------------------------------------------------------------------------------- /projects/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/README.md -------------------------------------------------------------------------------- /projects/align_detr/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/align_detr/README.md -------------------------------------------------------------------------------- /projects/align_detr/configs/aligndetr_k=2_r50_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/align_detr/configs/aligndetr_k=2_r50_4scale_12ep.py -------------------------------------------------------------------------------- /projects/align_detr/configs/aligndetr_k=2_r50_4scale_24ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/align_detr/configs/aligndetr_k=2_r50_4scale_24ep.py -------------------------------------------------------------------------------- /projects/align_detr/configs/aligndetr_k=2_r50_4scale_36ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/align_detr/configs/aligndetr_k=2_r50_4scale_36ep.py -------------------------------------------------------------------------------- /projects/align_detr/configs/models/aligndetr_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/align_detr/configs/models/aligndetr_r50.py -------------------------------------------------------------------------------- /projects/align_detr/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/align_detr/modeling/__init__.py -------------------------------------------------------------------------------- /projects/align_detr/modeling/aligndetr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/align_detr/modeling/aligndetr.py -------------------------------------------------------------------------------- /projects/align_detr/modeling/criterions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/align_detr/modeling/criterions/__init__.py -------------------------------------------------------------------------------- /projects/align_detr/modeling/criterions/aligndetr_dn_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/align_detr/modeling/criterions/aligndetr_dn_criterion.py -------------------------------------------------------------------------------- /projects/align_detr/modeling/criterions/base_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/align_detr/modeling/criterions/base_criterion.py -------------------------------------------------------------------------------- /projects/align_detr/modeling/criterions/many_to_one_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/align_detr/modeling/criterions/many_to_one_criterion.py -------------------------------------------------------------------------------- /projects/align_detr/modeling/criterions/two_stage_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/align_detr/modeling/criterions/two_stage_criterion.py -------------------------------------------------------------------------------- /projects/align_detr/modeling/losses/__init__.py: -------------------------------------------------------------------------------- 1 | from .losses import * -------------------------------------------------------------------------------- /projects/align_detr/modeling/losses/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/align_detr/modeling/losses/losses.py -------------------------------------------------------------------------------- /projects/align_detr/modeling/matchers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/align_detr/modeling/matchers/__init__.py -------------------------------------------------------------------------------- /projects/align_detr/modeling/matchers/mixed_matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/align_detr/modeling/matchers/mixed_matcher.py -------------------------------------------------------------------------------- /projects/align_detr/modeling/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/align_detr/modeling/transformer.py -------------------------------------------------------------------------------- /projects/anchor_detr/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/anchor_detr/README.md -------------------------------------------------------------------------------- /projects/anchor_detr/assets/anchor_detr_arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/anchor_detr/assets/anchor_detr_arch.png -------------------------------------------------------------------------------- /projects/anchor_detr/configs/anchor_detr_r101_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/anchor_detr/configs/anchor_detr_r101_50ep.py -------------------------------------------------------------------------------- /projects/anchor_detr/configs/anchor_detr_r101_dc5_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/anchor_detr/configs/anchor_detr_r101_dc5_50ep.py -------------------------------------------------------------------------------- /projects/anchor_detr/configs/anchor_detr_r50_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/anchor_detr/configs/anchor_detr_r50_50ep.py -------------------------------------------------------------------------------- /projects/anchor_detr/configs/anchor_detr_r50_dc5_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/anchor_detr/configs/anchor_detr_r50_dc5_50ep.py -------------------------------------------------------------------------------- /projects/anchor_detr/configs/models/anchor_detr_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/anchor_detr/configs/models/anchor_detr_r50.py -------------------------------------------------------------------------------- /projects/anchor_detr/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/anchor_detr/modeling/__init__.py -------------------------------------------------------------------------------- /projects/anchor_detr/modeling/anchor_detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/anchor_detr/modeling/anchor_detr.py -------------------------------------------------------------------------------- /projects/anchor_detr/modeling/anchor_detr_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/anchor_detr/modeling/anchor_detr_transformer.py -------------------------------------------------------------------------------- /projects/anchor_detr/modeling/row_column_decoupled_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/anchor_detr/modeling/row_column_decoupled_attention.py -------------------------------------------------------------------------------- /projects/anchor_detr/modeling/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/anchor_detr/modeling/utils.py -------------------------------------------------------------------------------- /projects/co_mot/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/co_mot/README.md -------------------------------------------------------------------------------- /projects/co_mot/configs/common/dancetrack_schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/co_mot/configs/common/dancetrack_schedule.py -------------------------------------------------------------------------------- /projects/co_mot/configs/common/data/dancetrack_mot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/co_mot/configs/common/data/dancetrack_mot.py -------------------------------------------------------------------------------- /projects/co_mot/configs/mot_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/co_mot/configs/mot_r50.py -------------------------------------------------------------------------------- /projects/co_mot/configs/mot_r50_4scale_10ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/co_mot/configs/mot_r50_4scale_10ep.py -------------------------------------------------------------------------------- /projects/co_mot/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/co_mot/data/__init__.py -------------------------------------------------------------------------------- /projects/co_mot/data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/co_mot/data/datasets/__init__.py -------------------------------------------------------------------------------- /projects/co_mot/data/datasets/register_dancetrack_mot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/co_mot/data/datasets/register_dancetrack_mot.py -------------------------------------------------------------------------------- /projects/co_mot/data/mot_build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/co_mot/data/mot_build.py -------------------------------------------------------------------------------- /projects/co_mot/data/mot_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/co_mot/data/mot_dataset_mapper.py -------------------------------------------------------------------------------- /projects/co_mot/data/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/co_mot/data/transforms/__init__.py -------------------------------------------------------------------------------- /projects/co_mot/data/transforms/mot_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/co_mot/data/transforms/mot_transforms.py -------------------------------------------------------------------------------- /projects/co_mot/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/co_mot/evaluation/__init__.py -------------------------------------------------------------------------------- /projects/co_mot/evaluation/dancetrack_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/co_mot/evaluation/dancetrack_evaluation.py -------------------------------------------------------------------------------- /projects/co_mot/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/co_mot/modeling/__init__.py -------------------------------------------------------------------------------- /projects/co_mot/modeling/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/co_mot/modeling/matcher.py -------------------------------------------------------------------------------- /projects/co_mot/modeling/mot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/co_mot/modeling/mot.py -------------------------------------------------------------------------------- /projects/co_mot/modeling/mot_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/co_mot/modeling/mot_transformer.py -------------------------------------------------------------------------------- /projects/co_mot/modeling/qim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/co_mot/modeling/qim.py -------------------------------------------------------------------------------- /projects/co_mot/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/co_mot/train_net.py -------------------------------------------------------------------------------- /projects/co_mot/util/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/co_mot/util/__init__.py -------------------------------------------------------------------------------- /projects/co_mot/util/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/co_mot/util/checkpoint.py -------------------------------------------------------------------------------- /projects/co_mot/util/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/co_mot/util/misc.py -------------------------------------------------------------------------------- /projects/conditional_detr/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/conditional_detr/README.md -------------------------------------------------------------------------------- /projects/conditional_detr/assets/attention-maps.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/conditional_detr/assets/attention-maps.png -------------------------------------------------------------------------------- /projects/conditional_detr/configs/conditional_detr_r101_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/conditional_detr/configs/conditional_detr_r101_50ep.py -------------------------------------------------------------------------------- /projects/conditional_detr/configs/conditional_detr_r101_dc5_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/conditional_detr/configs/conditional_detr_r101_dc5_50ep.py -------------------------------------------------------------------------------- /projects/conditional_detr/configs/conditional_detr_r50_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/conditional_detr/configs/conditional_detr_r50_50ep.py -------------------------------------------------------------------------------- /projects/conditional_detr/configs/conditional_detr_r50_dc5_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/conditional_detr/configs/conditional_detr_r50_dc5_50ep.py -------------------------------------------------------------------------------- /projects/conditional_detr/configs/models/conditional_detr_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/conditional_detr/configs/models/conditional_detr_r50.py -------------------------------------------------------------------------------- /projects/conditional_detr/configs/models/conditional_detr_r50_dc5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/conditional_detr/configs/models/conditional_detr_r50_dc5.py -------------------------------------------------------------------------------- /projects/conditional_detr/converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/conditional_detr/converter.py -------------------------------------------------------------------------------- /projects/conditional_detr/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/conditional_detr/modeling/__init__.py -------------------------------------------------------------------------------- /projects/conditional_detr/modeling/conditional_detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/conditional_detr/modeling/conditional_detr.py -------------------------------------------------------------------------------- /projects/conditional_detr/modeling/conditional_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/conditional_detr/modeling/conditional_transformer.py -------------------------------------------------------------------------------- /projects/dab_deformable_detr/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_deformable_detr/README.md -------------------------------------------------------------------------------- /projects/dab_deformable_detr/assets/dab_detr_overall.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_deformable_detr/assets/dab_detr_overall.png -------------------------------------------------------------------------------- /projects/dab_deformable_detr/configs/dab_deformable_detr_r50_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_deformable_detr/configs/dab_deformable_detr_r50_50ep.py -------------------------------------------------------------------------------- /projects/dab_deformable_detr/configs/dab_deformable_detr_r50_two_stage_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_deformable_detr/configs/dab_deformable_detr_r50_two_stage_50ep.py -------------------------------------------------------------------------------- /projects/dab_deformable_detr/configs/models/dab_deformable_detr_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_deformable_detr/configs/models/dab_deformable_detr_r50.py -------------------------------------------------------------------------------- /projects/dab_deformable_detr/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_deformable_detr/modeling/__init__.py -------------------------------------------------------------------------------- /projects/dab_deformable_detr/modeling/dab_deformable_detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_deformable_detr/modeling/dab_deformable_detr.py -------------------------------------------------------------------------------- /projects/dab_deformable_detr/modeling/dab_deformable_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_deformable_detr/modeling/dab_deformable_transformer.py -------------------------------------------------------------------------------- /projects/dab_deformable_detr/modeling/two_stage_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_deformable_detr/modeling/two_stage_criterion.py -------------------------------------------------------------------------------- /projects/dab_detr/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_detr/README.md -------------------------------------------------------------------------------- /projects/dab_detr/assets/dab_detr_details.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_detr/assets/dab_detr_details.png -------------------------------------------------------------------------------- /projects/dab_detr/assets/dab_detr_overall.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_detr/assets/dab_detr_overall.png -------------------------------------------------------------------------------- /projects/dab_detr/configs/dab_detr_r101_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_detr/configs/dab_detr_r101_50ep.py -------------------------------------------------------------------------------- /projects/dab_detr/configs/dab_detr_r101_dc5_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_detr/configs/dab_detr_r101_dc5_50ep.py -------------------------------------------------------------------------------- /projects/dab_detr/configs/dab_detr_r50_3patterns_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_detr/configs/dab_detr_r50_3patterns_50ep.py -------------------------------------------------------------------------------- /projects/dab_detr/configs/dab_detr_r50_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_detr/configs/dab_detr_r50_50ep.py -------------------------------------------------------------------------------- /projects/dab_detr/configs/dab_detr_r50_dc5_3patterns_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_detr/configs/dab_detr_r50_dc5_3patterns_50ep.py -------------------------------------------------------------------------------- /projects/dab_detr/configs/dab_detr_r50_dc5_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_detr/configs/dab_detr_r50_dc5_50ep.py -------------------------------------------------------------------------------- /projects/dab_detr/configs/dab_detr_swin_b_in21k_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_detr/configs/dab_detr_swin_b_in21k_50ep.py -------------------------------------------------------------------------------- /projects/dab_detr/configs/dab_detr_swin_t_in1k_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_detr/configs/dab_detr_swin_t_in1k_50ep.py -------------------------------------------------------------------------------- /projects/dab_detr/configs/models/dab_detr_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_detr/configs/models/dab_detr_r50.py -------------------------------------------------------------------------------- /projects/dab_detr/configs/models/dab_detr_r50_3patterns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_detr/configs/models/dab_detr_r50_3patterns.py -------------------------------------------------------------------------------- /projects/dab_detr/configs/models/dab_detr_r50_dc5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_detr/configs/models/dab_detr_r50_dc5.py -------------------------------------------------------------------------------- /projects/dab_detr/configs/models/dab_detr_swin_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_detr/configs/models/dab_detr_swin_base.py -------------------------------------------------------------------------------- /projects/dab_detr/configs/models/dab_detr_swin_tiny.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_detr/configs/models/dab_detr_swin_tiny.py -------------------------------------------------------------------------------- /projects/dab_detr/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_detr/modeling/__init__.py -------------------------------------------------------------------------------- /projects/dab_detr/modeling/dab_detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_detr/modeling/dab_detr.py -------------------------------------------------------------------------------- /projects/dab_detr/modeling/dab_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dab_detr/modeling/dab_transformer.py -------------------------------------------------------------------------------- /projects/deformable_detr/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deformable_detr/README.md -------------------------------------------------------------------------------- /projects/deformable_detr/assets/deformable_detr.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deformable_detr/assets/deformable_detr.png -------------------------------------------------------------------------------- /projects/deformable_detr/configs/deformable_detr_r50_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deformable_detr/configs/deformable_detr_r50_50ep.py -------------------------------------------------------------------------------- /projects/deformable_detr/configs/deformable_detr_r50_two_stage_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deformable_detr/configs/deformable_detr_r50_two_stage_50ep.py -------------------------------------------------------------------------------- /projects/deformable_detr/configs/deformable_detr_r50_with_box_refinement_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deformable_detr/configs/deformable_detr_r50_with_box_refinement_50ep.py -------------------------------------------------------------------------------- /projects/deformable_detr/configs/models/deformable_detr_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deformable_detr/configs/models/deformable_detr_r50.py -------------------------------------------------------------------------------- /projects/deformable_detr/convert_two_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deformable_detr/convert_two_stage.py -------------------------------------------------------------------------------- /projects/deformable_detr/converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deformable_detr/converter.py -------------------------------------------------------------------------------- /projects/deformable_detr/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deformable_detr/modeling/__init__.py -------------------------------------------------------------------------------- /projects/deformable_detr/modeling/deformable_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deformable_detr/modeling/deformable_criterion.py -------------------------------------------------------------------------------- /projects/deformable_detr/modeling/deformable_detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deformable_detr/modeling/deformable_detr.py -------------------------------------------------------------------------------- /projects/deformable_detr/modeling/deformable_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deformable_detr/modeling/deformable_transformer.py -------------------------------------------------------------------------------- /projects/deformable_detr/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deformable_detr/train_net.py -------------------------------------------------------------------------------- /projects/deta/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deta/README.md -------------------------------------------------------------------------------- /projects/deta/assets/deta.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deta/assets/deta.png -------------------------------------------------------------------------------- /projects/deta/configs/data/coco_detr_larger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deta/configs/data/coco_detr_larger.py -------------------------------------------------------------------------------- /projects/deta/configs/deta_r50_5scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deta/configs/deta_r50_5scale_12ep.py -------------------------------------------------------------------------------- /projects/deta/configs/deta_r50_5scale_12ep_bs8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deta/configs/deta_r50_5scale_12ep_bs8.py -------------------------------------------------------------------------------- /projects/deta/configs/deta_r50_5scale_no_frozen_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deta/configs/deta_r50_5scale_no_frozen_backbone.py -------------------------------------------------------------------------------- /projects/deta/configs/deta_swin_large_finetune_24ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deta/configs/deta_swin_large_finetune_24ep.py -------------------------------------------------------------------------------- /projects/deta/configs/improved_deformable_detr_baseline_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deta/configs/improved_deformable_detr_baseline_50ep.py -------------------------------------------------------------------------------- /projects/deta/configs/models/deta_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deta/configs/models/deta_r50.py -------------------------------------------------------------------------------- /projects/deta/configs/models/deta_swin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deta/configs/models/deta_swin.py -------------------------------------------------------------------------------- /projects/deta/configs/scheduler/coco_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deta/configs/scheduler/coco_scheduler.py -------------------------------------------------------------------------------- /projects/deta/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deta/modeling/__init__.py -------------------------------------------------------------------------------- /projects/deta/modeling/assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deta/modeling/assigner.py -------------------------------------------------------------------------------- /projects/deta/modeling/deformable_detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deta/modeling/deformable_detr.py -------------------------------------------------------------------------------- /projects/deta/modeling/deformable_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deta/modeling/deformable_transformer.py -------------------------------------------------------------------------------- /projects/deta/modeling/deta_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deta/modeling/deta_criterion.py -------------------------------------------------------------------------------- /projects/deta/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/deta/train_net.py -------------------------------------------------------------------------------- /projects/detr/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/detr/README.md -------------------------------------------------------------------------------- /projects/detr/assets/DETR.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/detr/assets/DETR.png -------------------------------------------------------------------------------- /projects/detr/configs/detr_r101_300ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/detr/configs/detr_r101_300ep.py -------------------------------------------------------------------------------- /projects/detr/configs/detr_r101_dc5_300ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/detr/configs/detr_r101_dc5_300ep.py -------------------------------------------------------------------------------- /projects/detr/configs/detr_r50_300ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/detr/configs/detr_r50_300ep.py -------------------------------------------------------------------------------- /projects/detr/configs/detr_r50_dc5_300ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/detr/configs/detr_r50_dc5_300ep.py -------------------------------------------------------------------------------- /projects/detr/configs/models/detr_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/detr/configs/models/detr_r50.py -------------------------------------------------------------------------------- /projects/detr/configs/models/detr_r50_dc5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/detr/configs/models/detr_r50_dc5.py -------------------------------------------------------------------------------- /projects/detr/converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/detr/converter.py -------------------------------------------------------------------------------- /projects/detr/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/detr/modeling/__init__.py -------------------------------------------------------------------------------- /projects/detr/modeling/detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/detr/modeling/detr.py -------------------------------------------------------------------------------- /projects/detr/modeling/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/detr/modeling/transformer.py -------------------------------------------------------------------------------- /projects/dino/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/README.md -------------------------------------------------------------------------------- /projects/dino/assets/dino_arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/assets/dino_arch.png -------------------------------------------------------------------------------- /projects/dino/configs/dino-convnext/dino_convnext_base_384_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-convnext/dino_convnext_base_384_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-convnext/dino_convnext_large_384_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-convnext/dino_convnext_large_384_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-convnext/dino_convnext_small_384_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-convnext/dino_convnext_small_384_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-convnext/dino_convnext_tiny_384_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-convnext/dino_convnext_tiny_384_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-eva-01/dino_eva_01_1536_4scale_12ep.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /projects/dino/configs/dino-eva-01/dino_eva_01_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-eva-01/dino_eva_01_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-focal/dino_focal_base_lrf_fl3_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-focal/dino_focal_base_lrf_fl3_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-focal/dino_focal_small_lrf_fl3_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-focal/dino_focal_small_lrf_fl3_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-focal/dino_focal_tiny_lrf_fl3_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-focal/dino_focal_tiny_lrf_fl3_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-focal/dino_focalnet_large_lrf_384_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-focal/dino_focalnet_large_lrf_384_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-focal/dino_focalnet_large_lrf_384_4scale_36ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-focal/dino_focalnet_large_lrf_384_4scale_36ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-focal/dino_focalnet_large_lrf_384_fl4_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-focal/dino_focalnet_large_lrf_384_fl4_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-focal/dino_focalnet_large_lrf_384_fl4_5scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-focal/dino_focalnet_large_lrf_384_fl4_5scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-focal/dino_focalnet_large_lrf_384_fl4_5scale_36ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-focal/dino_focalnet_large_lrf_384_fl4_5scale_36ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-focal/focalnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-focal/focalnet.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-internimage/dino_internimage_base_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-internimage/dino_internimage_base_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-internimage/dino_internimage_large_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-internimage/dino_internimage_large_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-internimage/dino_internimage_small_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-internimage/dino_internimage_small_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-internimage/dino_internimage_tiny_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-internimage/dino_internimage_tiny_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-resnet/dino_r101_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-resnet/dino_r101_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-resnet/dino_r50_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-resnet/dino_r50_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-resnet/dino_r50_4scale_12ep_300dn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-resnet/dino_r50_4scale_12ep_300dn.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-resnet/dino_r50_4scale_12ep_better_hyper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-resnet/dino_r50_4scale_12ep_better_hyper.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-resnet/dino_r50_4scale_12ep_no_frozen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-resnet/dino_r50_4scale_12ep_no_frozen.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-resnet/dino_r50_4scale_24ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-resnet/dino_r50_4scale_24ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-resnet/dino_r50_5scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-resnet/dino_r50_5scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-swin/dino_swin_base_384_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-swin/dino_swin_base_384_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-swin/dino_swin_large_224_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-swin/dino_swin_large_224_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-swin/dino_swin_large_384_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-swin/dino_swin_large_384_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-swin/dino_swin_large_384_4scale_36ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-swin/dino_swin_large_384_4scale_36ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-swin/dino_swin_large_384_5scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-swin/dino_swin_large_384_5scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-swin/dino_swin_large_384_5scale_36ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-swin/dino_swin_large_384_5scale_36ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-swin/dino_swin_small_224_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-swin/dino_swin_small_224_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-swin/dino_swin_tiny_224_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-swin/dino_swin_tiny_224_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-vitdet/dino_vitdet_base_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-vitdet/dino_vitdet_base_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-vitdet/dino_vitdet_base_4scale_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-vitdet/dino_vitdet_base_4scale_50ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-vitdet/dino_vitdet_large_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-vitdet/dino_vitdet_large_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino/configs/dino-vitdet/dino_vitdet_large_4scale_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/dino-vitdet/dino_vitdet_large_4scale_50ep.py -------------------------------------------------------------------------------- /projects/dino/configs/models/dino_convnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/models/dino_convnext.py -------------------------------------------------------------------------------- /projects/dino/configs/models/dino_eva_01.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/models/dino_eva_01.py -------------------------------------------------------------------------------- /projects/dino/configs/models/dino_focalnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/models/dino_focalnet.py -------------------------------------------------------------------------------- /projects/dino/configs/models/dino_internimage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/models/dino_internimage.py -------------------------------------------------------------------------------- /projects/dino/configs/models/dino_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/models/dino_r50.py -------------------------------------------------------------------------------- /projects/dino/configs/models/dino_swin_base_384.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/models/dino_swin_base_384.py -------------------------------------------------------------------------------- /projects/dino/configs/models/dino_swin_large_224.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/models/dino_swin_large_224.py -------------------------------------------------------------------------------- /projects/dino/configs/models/dino_swin_large_384.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/models/dino_swin_large_384.py -------------------------------------------------------------------------------- /projects/dino/configs/models/dino_swin_small_224.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/models/dino_swin_small_224.py -------------------------------------------------------------------------------- /projects/dino/configs/models/dino_swin_tiny_224.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/models/dino_swin_tiny_224.py -------------------------------------------------------------------------------- /projects/dino/configs/models/dino_vitdet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/models/dino_vitdet.py -------------------------------------------------------------------------------- /projects/dino/configs/timm_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/timm_example.py -------------------------------------------------------------------------------- /projects/dino/configs/torchvision_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/configs/torchvision_example.py -------------------------------------------------------------------------------- /projects/dino/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/modeling/__init__.py -------------------------------------------------------------------------------- /projects/dino/modeling/dino.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/modeling/dino.py -------------------------------------------------------------------------------- /projects/dino/modeling/dino_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/modeling/dino_transformer.py -------------------------------------------------------------------------------- /projects/dino/modeling/dn_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/modeling/dn_criterion.py -------------------------------------------------------------------------------- /projects/dino/modeling/two_stage_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/modeling/two_stage_criterion.py -------------------------------------------------------------------------------- /projects/dino/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino/train_net.py -------------------------------------------------------------------------------- /projects/dino_eva/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino_eva/README.md -------------------------------------------------------------------------------- /projects/dino_eva/assets/dino_arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino_eva/assets/dino_arch.png -------------------------------------------------------------------------------- /projects/dino_eva/configs/common/coco_loader_lsj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino_eva/configs/common/coco_loader_lsj.py -------------------------------------------------------------------------------- /projects/dino_eva/configs/common/coco_loader_lsj_1024.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino_eva/configs/common/coco_loader_lsj_1024.py -------------------------------------------------------------------------------- /projects/dino_eva/configs/common/coco_loader_lsj_1280.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino_eva/configs/common/coco_loader_lsj_1280.py -------------------------------------------------------------------------------- /projects/dino_eva/configs/common/coco_loader_lsj_1536.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino_eva/configs/common/coco_loader_lsj_1536.py -------------------------------------------------------------------------------- /projects/dino_eva/configs/dino-eva-01/dino_eva_01_1280_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino_eva/configs/dino-eva-01/dino_eva_01_1280_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino_eva/configs/dino-eva-01/dino_eva_01_1536_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino_eva/configs/dino-eva-01/dino_eva_01_1536_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino_eva/configs/dino-eva-02/dino_eva_02_vitdet_b_4attn_1024_lrd0p7_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino_eva/configs/dino-eva-02/dino_eva_02_vitdet_b_4attn_1024_lrd0p7_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino_eva/configs/dino-eva-02/dino_eva_02_vitdet_b_6attn_win32_1536_lrd0p7_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino_eva/configs/dino-eva-02/dino_eva_02_vitdet_b_6attn_win32_1536_lrd0p7_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino_eva/configs/dino-eva-02/dino_eva_02_vitdet_l_4attn_1024_lrd0p8_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino_eva/configs/dino-eva-02/dino_eva_02_vitdet_l_4attn_1024_lrd0p8_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino_eva/configs/dino-eva-02/dino_eva_02_vitdet_l_8attn_1536_lrd0p8_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino_eva/configs/dino-eva-02/dino_eva_02_vitdet_l_8attn_1536_lrd0p8_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino_eva/configs/dino-eva-02/dino_eva_02_vitdet_l_8attn_win32_1536_lrd0p8_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino_eva/configs/dino-eva-02/dino_eva_02_vitdet_l_8attn_win32_1536_lrd0p8_4scale_12ep.py -------------------------------------------------------------------------------- /projects/dino_eva/configs/models/dino_eva_01.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino_eva/configs/models/dino_eva_01.py -------------------------------------------------------------------------------- /projects/dino_eva/configs/models/dino_eva_02.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino_eva/configs/models/dino_eva_02.py -------------------------------------------------------------------------------- /projects/dino_eva/configs/models/dino_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino_eva/configs/models/dino_r50.py -------------------------------------------------------------------------------- /projects/dino_eva/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino_eva/modeling/__init__.py -------------------------------------------------------------------------------- /projects/dino_eva/modeling/dino.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino_eva/modeling/dino.py -------------------------------------------------------------------------------- /projects/dino_eva/modeling/dino_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino_eva/modeling/dino_transformer.py -------------------------------------------------------------------------------- /projects/dino_eva/modeling/dn_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino_eva/modeling/dn_criterion.py -------------------------------------------------------------------------------- /projects/dino_eva/modeling/two_stage_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino_eva/modeling/two_stage_criterion.py -------------------------------------------------------------------------------- /projects/dino_eva/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dino_eva/train_net.py -------------------------------------------------------------------------------- /projects/dn_deformable_detr/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dn_deformable_detr/README.md -------------------------------------------------------------------------------- /projects/dn_deformable_detr/assets/dn_detr_arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dn_deformable_detr/assets/dn_detr_arch.png -------------------------------------------------------------------------------- /projects/dn_deformable_detr/configs/dn_deformable_detr_r50_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dn_deformable_detr/configs/dn_deformable_detr_r50_12ep.py -------------------------------------------------------------------------------- /projects/dn_deformable_detr/configs/dn_deformable_detr_r50_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dn_deformable_detr/configs/dn_deformable_detr_r50_50ep.py -------------------------------------------------------------------------------- /projects/dn_deformable_detr/configs/models/dn_deformable_detr_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dn_deformable_detr/configs/models/dn_deformable_detr_r50.py -------------------------------------------------------------------------------- /projects/dn_deformable_detr/converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dn_deformable_detr/converter.py -------------------------------------------------------------------------------- /projects/dn_deformable_detr/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dn_deformable_detr/modeling/__init__.py -------------------------------------------------------------------------------- /projects/dn_deformable_detr/modeling/dn_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dn_deformable_detr/modeling/dn_criterion.py -------------------------------------------------------------------------------- /projects/dn_deformable_detr/modeling/dn_deformable_detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dn_deformable_detr/modeling/dn_deformable_detr.py -------------------------------------------------------------------------------- /projects/dn_deformable_detr/modeling/dn_deformable_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dn_deformable_detr/modeling/dn_deformable_transformer.py -------------------------------------------------------------------------------- /projects/dn_detr/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dn_detr/README.md -------------------------------------------------------------------------------- /projects/dn_detr/assets/dn_detr_arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dn_detr/assets/dn_detr_arch.png -------------------------------------------------------------------------------- /projects/dn_detr/configs/dn_detr_r50_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dn_detr/configs/dn_detr_r50_50ep.py -------------------------------------------------------------------------------- /projects/dn_detr/configs/dn_detr_r50_dc5_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dn_detr/configs/dn_detr_r50_dc5_50ep.py -------------------------------------------------------------------------------- /projects/dn_detr/configs/models/dn_detr_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dn_detr/configs/models/dn_detr_r50.py -------------------------------------------------------------------------------- /projects/dn_detr/configs/models/dn_detr_r50_dc5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dn_detr/configs/models/dn_detr_r50_dc5.py -------------------------------------------------------------------------------- /projects/dn_detr/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dn_detr/modeling/__init__.py -------------------------------------------------------------------------------- /projects/dn_detr/modeling/dn_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dn_detr/modeling/dn_criterion.py -------------------------------------------------------------------------------- /projects/dn_detr/modeling/dn_detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dn_detr/modeling/dn_detr.py -------------------------------------------------------------------------------- /projects/dn_detr/modeling/dn_transformers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/dn_detr/modeling/dn_transformers.py -------------------------------------------------------------------------------- /projects/focus_detr/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/README.md -------------------------------------------------------------------------------- /projects/focus_detr/configs/focus_detr_resnet/focus_detr_r101_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/configs/focus_detr_resnet/focus_detr_r101_4scale_12ep.py -------------------------------------------------------------------------------- /projects/focus_detr/configs/focus_detr_resnet/focus_detr_r101_4scale_24ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/configs/focus_detr_resnet/focus_detr_r101_4scale_24ep.py -------------------------------------------------------------------------------- /projects/focus_detr/configs/focus_detr_resnet/focus_detr_r101_4scale_36ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/configs/focus_detr_resnet/focus_detr_r101_4scale_36ep.py -------------------------------------------------------------------------------- /projects/focus_detr/configs/focus_detr_resnet/focus_detr_r50_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/configs/focus_detr_resnet/focus_detr_r50_4scale_12ep.py -------------------------------------------------------------------------------- /projects/focus_detr/configs/focus_detr_resnet/focus_detr_r50_4scale_24ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/configs/focus_detr_resnet/focus_detr_r50_4scale_24ep.py -------------------------------------------------------------------------------- /projects/focus_detr/configs/focus_detr_resnet/focus_detr_r50_4scale_36ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/configs/focus_detr_resnet/focus_detr_r50_4scale_36ep.py -------------------------------------------------------------------------------- /projects/focus_detr/configs/focus_detr_swin/focus_detr_swin_base_224_4scale_36ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/configs/focus_detr_swin/focus_detr_swin_base_224_4scale_36ep.py -------------------------------------------------------------------------------- /projects/focus_detr/configs/focus_detr_swin/focus_detr_swin_base_384_4scale_36ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/configs/focus_detr_swin/focus_detr_swin_base_384_4scale_36ep.py -------------------------------------------------------------------------------- /projects/focus_detr/configs/focus_detr_swin/focus_detr_swin_large_384_4scale_36ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/configs/focus_detr_swin/focus_detr_swin_large_384_4scale_36ep.py -------------------------------------------------------------------------------- /projects/focus_detr/configs/focus_detr_swin/focus_detr_swin_tiny_224_4scale_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/configs/focus_detr_swin/focus_detr_swin_tiny_224_4scale_12ep.py -------------------------------------------------------------------------------- /projects/focus_detr/configs/focus_detr_swin/focus_detr_swin_tiny_224_4scale_22k_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/configs/focus_detr_swin/focus_detr_swin_tiny_224_4scale_22k_12ep.py -------------------------------------------------------------------------------- /projects/focus_detr/configs/focus_detr_swin/focus_detr_swin_tiny_224_4scale_22k_36ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/configs/focus_detr_swin/focus_detr_swin_tiny_224_4scale_22k_36ep.py -------------------------------------------------------------------------------- /projects/focus_detr/configs/focus_detr_swin/focus_detr_swin_tiny_224_4scale_24ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/configs/focus_detr_swin/focus_detr_swin_tiny_224_4scale_24ep.py -------------------------------------------------------------------------------- /projects/focus_detr/configs/focus_detr_swin/focus_detr_swin_tiny_224_4scale_36ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/configs/focus_detr_swin/focus_detr_swin_tiny_224_4scale_36ep.py -------------------------------------------------------------------------------- /projects/focus_detr/configs/models/focus_detr_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/configs/models/focus_detr_r50.py -------------------------------------------------------------------------------- /projects/focus_detr/configs/models/focus_detr_swin_base_384.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/configs/models/focus_detr_swin_base_384.py -------------------------------------------------------------------------------- /projects/focus_detr/configs/models/focus_detr_swin_large_224.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/configs/models/focus_detr_swin_large_224.py -------------------------------------------------------------------------------- /projects/focus_detr/configs/models/focus_detr_swin_large_384.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/configs/models/focus_detr_swin_large_384.py -------------------------------------------------------------------------------- /projects/focus_detr/configs/models/focus_detr_swin_small_224.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/configs/models/focus_detr_swin_small_224.py -------------------------------------------------------------------------------- /projects/focus_detr/configs/models/focus_detr_swin_tiny_224.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/configs/models/focus_detr_swin_tiny_224.py -------------------------------------------------------------------------------- /projects/focus_detr/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/modeling/__init__.py -------------------------------------------------------------------------------- /projects/focus_detr/modeling/dn_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/modeling/dn_criterion.py -------------------------------------------------------------------------------- /projects/focus_detr/modeling/focus_detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/modeling/focus_detr.py -------------------------------------------------------------------------------- /projects/focus_detr/modeling/focus_detr_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/modeling/focus_detr_transformer.py -------------------------------------------------------------------------------- /projects/focus_detr/modeling/foreground_supervision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/modeling/foreground_supervision.py -------------------------------------------------------------------------------- /projects/focus_detr/modeling/transformer_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/modeling/transformer_layer.py -------------------------------------------------------------------------------- /projects/focus_detr/modeling/two_stage_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/focus_detr/modeling/two_stage_criterion.py -------------------------------------------------------------------------------- /projects/group_detr/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/group_detr/README.md -------------------------------------------------------------------------------- /projects/group_detr/assets/group_detr_arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/group_detr/assets/group_detr_arch.png -------------------------------------------------------------------------------- /projects/group_detr/configs/group_detr_r50_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/group_detr/configs/group_detr_r50_50ep.py -------------------------------------------------------------------------------- /projects/group_detr/configs/models/group_detr_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/group_detr/configs/models/group_detr_r50.py -------------------------------------------------------------------------------- /projects/group_detr/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/group_detr/modeling/__init__.py -------------------------------------------------------------------------------- /projects/group_detr/modeling/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/group_detr/modeling/attention.py -------------------------------------------------------------------------------- /projects/group_detr/modeling/group_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/group_detr/modeling/group_criterion.py -------------------------------------------------------------------------------- /projects/group_detr/modeling/group_detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/group_detr/modeling/group_detr.py -------------------------------------------------------------------------------- /projects/group_detr/modeling/group_detr_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/group_detr/modeling/group_detr_transformer.py -------------------------------------------------------------------------------- /projects/group_detr/modeling/group_matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/group_detr/modeling/group_matcher.py -------------------------------------------------------------------------------- /projects/h_deformable_detr/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/h_deformable_detr/README.md -------------------------------------------------------------------------------- /projects/h_deformable_detr/assets/h_detr_arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/h_deformable_detr/assets/h_detr_arch.png -------------------------------------------------------------------------------- /projects/h_deformable_detr/configs/h_deformable_detr_r50_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/h_deformable_detr/configs/h_deformable_detr_r50_50ep.py -------------------------------------------------------------------------------- /projects/h_deformable_detr/configs/h_deformable_detr_r50_two_stage_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/h_deformable_detr/configs/h_deformable_detr_r50_two_stage_12ep.py -------------------------------------------------------------------------------- /projects/h_deformable_detr/configs/h_deformable_detr_r50_two_stage_36ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/h_deformable_detr/configs/h_deformable_detr_r50_two_stage_36ep.py -------------------------------------------------------------------------------- /projects/h_deformable_detr/configs/h_deformable_detr_swin_large_two_stage_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/h_deformable_detr/configs/h_deformable_detr_swin_large_two_stage_12ep.py -------------------------------------------------------------------------------- /projects/h_deformable_detr/configs/h_deformable_detr_swin_large_two_stage_12ep_900queries.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/h_deformable_detr/configs/h_deformable_detr_swin_large_two_stage_12ep_900queries.py -------------------------------------------------------------------------------- /projects/h_deformable_detr/configs/h_deformable_detr_swin_large_two_stage_36ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/h_deformable_detr/configs/h_deformable_detr_swin_large_two_stage_36ep.py -------------------------------------------------------------------------------- /projects/h_deformable_detr/configs/h_deformable_detr_swin_large_two_stage_36ep_900queries.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/h_deformable_detr/configs/h_deformable_detr_swin_large_two_stage_36ep_900queries.py -------------------------------------------------------------------------------- /projects/h_deformable_detr/configs/h_deformable_detr_swin_tiny_two_stage_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/h_deformable_detr/configs/h_deformable_detr_swin_tiny_two_stage_12ep.py -------------------------------------------------------------------------------- /projects/h_deformable_detr/configs/h_deformable_detr_swin_tiny_two_stage_36ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/h_deformable_detr/configs/h_deformable_detr_swin_tiny_two_stage_36ep.py -------------------------------------------------------------------------------- /projects/h_deformable_detr/configs/models/h_deformable_detr_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/h_deformable_detr/configs/models/h_deformable_detr_r50.py -------------------------------------------------------------------------------- /projects/h_deformable_detr/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/h_deformable_detr/modeling/__init__.py -------------------------------------------------------------------------------- /projects/h_deformable_detr/modeling/deformable_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/h_deformable_detr/modeling/deformable_criterion.py -------------------------------------------------------------------------------- /projects/h_deformable_detr/modeling/h_deformable_detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/h_deformable_detr/modeling/h_deformable_detr.py -------------------------------------------------------------------------------- /projects/h_deformable_detr/modeling/h_deformable_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/h_deformable_detr/modeling/h_deformable_transformer.py -------------------------------------------------------------------------------- /projects/h_deformable_detr/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/h_deformable_detr/train_net.py -------------------------------------------------------------------------------- /projects/maskdino/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/README.md -------------------------------------------------------------------------------- /projects/maskdino/assets/dinosaur.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/assets/dinosaur.png -------------------------------------------------------------------------------- /projects/maskdino/assets/framework.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/assets/framework.jpg -------------------------------------------------------------------------------- /projects/maskdino/assets/instance.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/assets/instance.png -------------------------------------------------------------------------------- /projects/maskdino/assets/panoptic.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/assets/panoptic.png -------------------------------------------------------------------------------- /projects/maskdino/assets/semantic.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/assets/semantic.png -------------------------------------------------------------------------------- /projects/maskdino/assets/sota.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/assets/sota.png -------------------------------------------------------------------------------- /projects/maskdino/configs/data/ade20k_semantic_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/configs/data/ade20k_semantic_seg.py -------------------------------------------------------------------------------- /projects/maskdino/configs/data/coco_instance_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/configs/data/coco_instance_seg.py -------------------------------------------------------------------------------- /projects/maskdino/configs/data/coco_panoptic_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/configs/data/coco_panoptic_seg.py -------------------------------------------------------------------------------- /projects/maskdino/configs/maskdino_r50_ade20k_semantic_seg_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/configs/maskdino_r50_ade20k_semantic_seg_160k.py -------------------------------------------------------------------------------- /projects/maskdino/configs/maskdino_r50_coco_instance_seg_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/configs/maskdino_r50_coco_instance_seg_50ep.py -------------------------------------------------------------------------------- /projects/maskdino/configs/maskdino_r50_coco_panoptic_seg_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/configs/maskdino_r50_coco_panoptic_seg_50ep.py -------------------------------------------------------------------------------- /projects/maskdino/configs/maskdino_r50_instance_seg_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/configs/maskdino_r50_instance_seg_50ep.py -------------------------------------------------------------------------------- /projects/maskdino/configs/models/maskdino_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/configs/models/maskdino_r50.py -------------------------------------------------------------------------------- /projects/maskdino/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/data/__init__.py -------------------------------------------------------------------------------- /projects/maskdino/data/dataset_mappers/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. 2 | -------------------------------------------------------------------------------- /projects/maskdino/data/dataset_mappers/coco_instance_lsj_aug_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/data/dataset_mappers/coco_instance_lsj_aug_dataset_mapper.py -------------------------------------------------------------------------------- /projects/maskdino/evaluation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /projects/maskdino/evaluation/instance_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/evaluation/instance_evaluation.py -------------------------------------------------------------------------------- /projects/maskdino/maskdino.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/maskdino.py -------------------------------------------------------------------------------- /projects/maskdino/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/modeling/__init__.py -------------------------------------------------------------------------------- /projects/maskdino/modeling/backbone/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. 2 | -------------------------------------------------------------------------------- /projects/maskdino/modeling/backbone/focal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/modeling/backbone/focal.py -------------------------------------------------------------------------------- /projects/maskdino/modeling/backbone/swin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/modeling/backbone/swin.py -------------------------------------------------------------------------------- /projects/maskdino/modeling/criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/modeling/criterion.py -------------------------------------------------------------------------------- /projects/maskdino/modeling/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/modeling/matcher.py -------------------------------------------------------------------------------- /projects/maskdino/modeling/meta_arch/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) IDEA, Inc. and its affiliates. 2 | 3 | -------------------------------------------------------------------------------- /projects/maskdino/modeling/meta_arch/maskdino_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/modeling/meta_arch/maskdino_head.py -------------------------------------------------------------------------------- /projects/maskdino/modeling/pixel_decoder/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) IDEA, Inc. and its affiliates. 2 | -------------------------------------------------------------------------------- /projects/maskdino/modeling/pixel_decoder/maskdino_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/modeling/pixel_decoder/maskdino_encoder.py -------------------------------------------------------------------------------- /projects/maskdino/modeling/pixel_decoder/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/modeling/pixel_decoder/position_encoding.py -------------------------------------------------------------------------------- /projects/maskdino/modeling/transformer_decoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/modeling/transformer_decoder/__init__.py -------------------------------------------------------------------------------- /projects/maskdino/modeling/transformer_decoder/dino_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/modeling/transformer_decoder/dino_decoder.py -------------------------------------------------------------------------------- /projects/maskdino/modeling/transformer_decoder/maskdino_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/modeling/transformer_decoder/maskdino_decoder.py -------------------------------------------------------------------------------- /projects/maskdino/modeling/transformer_decoder/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/modeling/transformer_decoder/utils.py -------------------------------------------------------------------------------- /projects/maskdino/utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. 2 | # import misc -------------------------------------------------------------------------------- /projects/maskdino/utils/box_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/utils/box_ops.py -------------------------------------------------------------------------------- /projects/maskdino/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/utils/misc.py -------------------------------------------------------------------------------- /projects/maskdino/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/maskdino/utils/utils.py -------------------------------------------------------------------------------- /projects/pnp_detr/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/pnp_detr/README.md -------------------------------------------------------------------------------- /projects/pnp_detr/assets/PnP-DETR.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/pnp_detr/assets/PnP-DETR.png -------------------------------------------------------------------------------- /projects/pnp_detr/configs/models/pnp_detr_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/pnp_detr/configs/models/pnp_detr_r50.py -------------------------------------------------------------------------------- /projects/pnp_detr/configs/pnp_detr_r101_300ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/pnp_detr/configs/pnp_detr_r101_300ep.py -------------------------------------------------------------------------------- /projects/pnp_detr/configs/pnp_detr_r50_300ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/pnp_detr/configs/pnp_detr_r50_300ep.py -------------------------------------------------------------------------------- /projects/pnp_detr/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/pnp_detr/modeling/__init__.py -------------------------------------------------------------------------------- /projects/pnp_detr/modeling/detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/pnp_detr/modeling/detr.py -------------------------------------------------------------------------------- /projects/pnp_detr/modeling/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/pnp_detr/modeling/transformer.py -------------------------------------------------------------------------------- /projects/rank_detr/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/rank_detr/README.md -------------------------------------------------------------------------------- /projects/rank_detr/assets/rank_detr_overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/rank_detr/assets/rank_detr_overview.png -------------------------------------------------------------------------------- /projects/rank_detr/configs/models/rank_detr_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/rank_detr/configs/models/rank_detr_r50.py -------------------------------------------------------------------------------- /projects/rank_detr/configs/rank_detr_r50_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/rank_detr/configs/rank_detr_r50_50ep.py -------------------------------------------------------------------------------- /projects/rank_detr/configs/rank_detr_r50_two_stage_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/rank_detr/configs/rank_detr_r50_two_stage_12ep.py -------------------------------------------------------------------------------- /projects/rank_detr/configs/rank_detr_r50_two_stage_36ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/rank_detr/configs/rank_detr_r50_two_stage_36ep.py -------------------------------------------------------------------------------- /projects/rank_detr/configs/rank_detr_swin_large_two_stage_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/rank_detr/configs/rank_detr_swin_large_two_stage_12ep.py -------------------------------------------------------------------------------- /projects/rank_detr/configs/rank_detr_swin_large_two_stage_36ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/rank_detr/configs/rank_detr_swin_large_two_stage_36ep.py -------------------------------------------------------------------------------- /projects/rank_detr/configs/rank_detr_swin_tiny_two_stage_12ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/rank_detr/configs/rank_detr_swin_tiny_two_stage_12ep.py -------------------------------------------------------------------------------- /projects/rank_detr/configs/rank_detr_swin_tiny_two_stage_36ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/rank_detr/configs/rank_detr_swin_tiny_two_stage_36ep.py -------------------------------------------------------------------------------- /projects/rank_detr/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/rank_detr/modeling/__init__.py -------------------------------------------------------------------------------- /projects/rank_detr/modeling/high_order_matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/rank_detr/modeling/high_order_matcher.py -------------------------------------------------------------------------------- /projects/rank_detr/modeling/rank_detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/rank_detr/modeling/rank_detr.py -------------------------------------------------------------------------------- /projects/rank_detr/modeling/rank_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/rank_detr/modeling/rank_transformer.py -------------------------------------------------------------------------------- /projects/rank_detr/modeling/rankdetr_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/rank_detr/modeling/rankdetr_criterion.py -------------------------------------------------------------------------------- /projects/rank_detr/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/rank_detr/train_net.py -------------------------------------------------------------------------------- /projects/sqr_detr/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/sqr_detr/README.md -------------------------------------------------------------------------------- /projects/sqr_detr/assets/sqr_detr_overall.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/sqr_detr/assets/sqr_detr_overall.png -------------------------------------------------------------------------------- /projects/sqr_detr/configs/dab_detr_r50_50ep_sqr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/sqr_detr/configs/dab_detr_r50_50ep_sqr.py -------------------------------------------------------------------------------- /projects/sqr_detr/configs/models/dab_detr_r50_sqr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/sqr_detr/configs/models/dab_detr_r50_sqr.py -------------------------------------------------------------------------------- /projects/sqr_detr/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/sqr_detr/modeling/__init__.py -------------------------------------------------------------------------------- /projects/sqr_detr/modeling/dab_transformer_sqr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/projects/sqr_detr/modeling/dab_transformer_sqr.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/setup.py -------------------------------------------------------------------------------- /tests/test_cond_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/tests/test_cond_attn.py -------------------------------------------------------------------------------- /tests/test_ffn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/tests/test_ffn.py -------------------------------------------------------------------------------- /tests/test_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/tests/test_losses.py -------------------------------------------------------------------------------- /tests/test_ms_deform_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/tests/test_ms_deform_attn.py -------------------------------------------------------------------------------- /tests/test_position_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/tests/test_position_embedding.py -------------------------------------------------------------------------------- /tests/test_torchvision_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/tests/test_torchvision_backbone.py -------------------------------------------------------------------------------- /tests/test_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/tests/test_transformer.py -------------------------------------------------------------------------------- /tests/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/tests/utils/__init__.py -------------------------------------------------------------------------------- /tests/utils/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/tests/utils/attention.py -------------------------------------------------------------------------------- /tests/utils/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/tests/utils/losses.py -------------------------------------------------------------------------------- /tests/utils/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/tests/utils/mlp.py -------------------------------------------------------------------------------- /tests/utils/potision_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/tests/utils/potision_embedding.py -------------------------------------------------------------------------------- /tests/utils/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/tests/utils/transformer.py -------------------------------------------------------------------------------- /tools/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/tools/README.md -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/analyze_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/tools/analyze_model.py -------------------------------------------------------------------------------- /tools/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/tools/benchmark.py -------------------------------------------------------------------------------- /tools/hydra_train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/tools/hydra_train_net.py -------------------------------------------------------------------------------- /tools/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/tools/train_net.py -------------------------------------------------------------------------------- /tools/visualize_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/tools/visualize_data.py -------------------------------------------------------------------------------- /tools/visualize_json_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeapLabTHU/Rank-DETR/HEAD/tools/visualize_json_results.py --------------------------------------------------------------------------------