├── .gitignore ├── LICENSE.txt ├── README.md ├── asserts ├── framework.jpg ├── imgs │ ├── Figure_1.jpg │ └── Figure_2.jpg ├── outdir │ ├── full_half_fruit_Figure_2.jpg │ └── three_skateboard_guys_Figure_1.jpg └── outdir1 │ └── three_skateboard_guys_Figure_1.jpg ├── configs ├── _base_ │ ├── datasets │ │ ├── gres │ │ │ ├── ref-zom.py │ │ │ └── rrefcoco.py │ │ └── segmentation │ │ │ ├── grefs.py │ │ │ ├── mixed-seg_nogoogle.py │ │ │ ├── refcoco-unc.py │ │ │ ├── refcocog-umd.py │ │ │ └── refcocoplus-unc.py │ └── misc.py ├── gres │ └── PropVG-grefcoco.py ├── refcoco+ │ └── PropVG-refcoco+.py ├── refcoco-mix │ └── PropVG-refcoco-mix.py ├── refcoco │ └── PropVG-refcoco.py ├── refcocog │ └── PropVG-refcocog.py ├── refzom │ └── PropVG-refzom.py └── rrefcoco │ └── PropVG-rrefcoco.py ├── data ├── multi_test.sh ├── pretrain_weights ├── propvg ├── __init__.py ├── apis │ ├── __init__.py │ ├── test.py │ └── train.py ├── core │ ├── __init__.py │ ├── criterion │ │ ├── __init__.py │ │ ├── base_criterion.py │ │ ├── criterion.py │ │ ├── distill_criterion.py │ │ ├── distill_cross_entropy_loss.py │ │ ├── distill_hungarian_assigner.py │ │ └── match_cost.py │ ├── layers │ │ ├── __init__.py │ │ ├── mdetr_transformer.py │ │ └── transformer.py │ ├── losses │ │ ├── __init__.py │ │ └── label_smooth.py │ ├── matcher.py │ ├── matcher │ │ ├── __init__.py │ │ ├── match_cost.py │ │ ├── matcher.py │ │ └── modified_matcher.py │ ├── optimizer.py │ ├── scheduler.py │ ├── structure │ │ ├── __init__.py │ │ ├── boxes.py │ │ ├── instances.py │ │ └── postprocessing.py │ └── utils.py ├── datasets │ ├── __init__.py │ ├── base.py │ ├── builder.py │ ├── pipelines │ │ ├── __init__.py │ │ ├── compose.py │ │ ├── formatting.py │ │ ├── loading.py │ │ ├── mask.py │ │ ├── transforms.py │ │ └── vgtr_aug.py │ ├── read.py │ ├── utils.py │ └── vgtr_dataset.py ├── evaluations │ ├── grefcoco.py │ ├── refzom.py │ └── rrefcoco.py ├── layers │ ├── __init__.py │ ├── attention.py │ ├── box_ops.py │ ├── conv.py │ ├── dcn_v3.py │ ├── denoising.py │ ├── layer_norm.py │ ├── mlp.py │ ├── multi_scale_deform_attn.py │ ├── position_embedding.py │ ├── shape_spec.py │ └── transformer.py ├── models │ ├── __init__.py │ ├── builder.py │ ├── det_seg │ │ ├── __init__.py │ │ ├── base.py │ │ ├── mix_gref_unimodel_OMG.py │ │ ├── mix_ref_unimodel_OMG.py │ │ └── one_stage.py │ ├── fusions │ │ ├── __init__.py │ │ ├── fusion.py │ │ ├── mdetr_fusion.py │ │ └── simple_fpn_fusion.py │ ├── heads │ │ ├── GTMHead.py │ │ ├── REFHead.py │ │ ├── __init__.py │ │ ├── modules.py │ │ ├── projection.py │ │ ├── tgqs_kd_detr_head │ │ │ ├── tgqs_kd_detr_head.py │ │ │ ├── transformer.py │ │ │ └── utils.py │ │ ├── transformers │ │ │ ├── baselayer_transformer.py │ │ │ ├── deformable_transformer.py │ │ │ ├── transformer.py │ │ │ └── utils.py │ │ └── unet_head.py │ ├── lan_encs │ │ ├── __init__.py │ │ ├── bert.py │ │ ├── lstm.py │ │ └── rnn.py │ ├── losses │ │ ├── __init__.py │ │ ├── boxloss.py │ │ ├── cem.py │ │ ├── clip_loss.py │ │ ├── contristiveloss.py │ │ └── segloss.py │ ├── postprocess │ │ ├── nls.py │ │ ├── nms.py │ │ └── nms_reverse.py │ ├── utils.py │ └── vis_encs │ │ ├── __init__.py │ │ ├── beit │ │ ├── beit3.py │ │ ├── beit3_base.py │ │ ├── modeling_finetune.py │ │ ├── modeling_utils.py │ │ └── utils.py │ │ ├── cspdarknet.py │ │ ├── darknet.py │ │ └── utils.py ├── utils │ ├── __init__.py │ ├── checkpoint.py │ ├── det_cam_visualizer.py │ ├── dist.py │ ├── distributed.py │ ├── gradcam_models.py │ ├── logger.py │ ├── visual_utils.py │ └── visualizer.py └── version.py ├── requirements.txt ├── setup.py └── tools ├── demo.py ├── dist_test.sh ├── dist_train.sh ├── inference.py ├── test.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/README.md -------------------------------------------------------------------------------- /asserts/framework.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/asserts/framework.jpg -------------------------------------------------------------------------------- /asserts/imgs/Figure_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/asserts/imgs/Figure_1.jpg -------------------------------------------------------------------------------- /asserts/imgs/Figure_2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/asserts/imgs/Figure_2.jpg -------------------------------------------------------------------------------- /asserts/outdir/full_half_fruit_Figure_2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/asserts/outdir/full_half_fruit_Figure_2.jpg -------------------------------------------------------------------------------- /asserts/outdir/three_skateboard_guys_Figure_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/asserts/outdir/three_skateboard_guys_Figure_1.jpg -------------------------------------------------------------------------------- /asserts/outdir1/three_skateboard_guys_Figure_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/asserts/outdir1/three_skateboard_guys_Figure_1.jpg -------------------------------------------------------------------------------- /configs/_base_/datasets/gres/ref-zom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/configs/_base_/datasets/gres/ref-zom.py -------------------------------------------------------------------------------- /configs/_base_/datasets/gres/rrefcoco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/configs/_base_/datasets/gres/rrefcoco.py -------------------------------------------------------------------------------- /configs/_base_/datasets/segmentation/grefs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/configs/_base_/datasets/segmentation/grefs.py -------------------------------------------------------------------------------- /configs/_base_/datasets/segmentation/mixed-seg_nogoogle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/configs/_base_/datasets/segmentation/mixed-seg_nogoogle.py -------------------------------------------------------------------------------- /configs/_base_/datasets/segmentation/refcoco-unc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/configs/_base_/datasets/segmentation/refcoco-unc.py -------------------------------------------------------------------------------- /configs/_base_/datasets/segmentation/refcocog-umd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/configs/_base_/datasets/segmentation/refcocog-umd.py -------------------------------------------------------------------------------- /configs/_base_/datasets/segmentation/refcocoplus-unc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/configs/_base_/datasets/segmentation/refcocoplus-unc.py -------------------------------------------------------------------------------- /configs/_base_/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/configs/_base_/misc.py -------------------------------------------------------------------------------- /configs/gres/PropVG-grefcoco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/configs/gres/PropVG-grefcoco.py -------------------------------------------------------------------------------- /configs/refcoco+/PropVG-refcoco+.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/configs/refcoco+/PropVG-refcoco+.py -------------------------------------------------------------------------------- /configs/refcoco-mix/PropVG-refcoco-mix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/configs/refcoco-mix/PropVG-refcoco-mix.py -------------------------------------------------------------------------------- /configs/refcoco/PropVG-refcoco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/configs/refcoco/PropVG-refcoco.py -------------------------------------------------------------------------------- /configs/refcocog/PropVG-refcocog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/configs/refcocog/PropVG-refcocog.py -------------------------------------------------------------------------------- /configs/refzom/PropVG-refzom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/configs/refzom/PropVG-refzom.py -------------------------------------------------------------------------------- /configs/rrefcoco/PropVG-rrefcoco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/configs/rrefcoco/PropVG-rrefcoco.py -------------------------------------------------------------------------------- /data: -------------------------------------------------------------------------------- 1 | ../SimVG/data -------------------------------------------------------------------------------- /multi_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/multi_test.sh -------------------------------------------------------------------------------- /pretrain_weights: -------------------------------------------------------------------------------- 1 | ../SimVG/pretrain_weights -------------------------------------------------------------------------------- /propvg/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /propvg/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/apis/__init__.py -------------------------------------------------------------------------------- /propvg/apis/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/apis/test.py -------------------------------------------------------------------------------- /propvg/apis/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/apis/train.py -------------------------------------------------------------------------------- /propvg/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/__init__.py -------------------------------------------------------------------------------- /propvg/core/criterion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/criterion/__init__.py -------------------------------------------------------------------------------- /propvg/core/criterion/base_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/criterion/base_criterion.py -------------------------------------------------------------------------------- /propvg/core/criterion/criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/criterion/criterion.py -------------------------------------------------------------------------------- /propvg/core/criterion/distill_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/criterion/distill_criterion.py -------------------------------------------------------------------------------- /propvg/core/criterion/distill_cross_entropy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/criterion/distill_cross_entropy_loss.py -------------------------------------------------------------------------------- /propvg/core/criterion/distill_hungarian_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/criterion/distill_hungarian_assigner.py -------------------------------------------------------------------------------- /propvg/core/criterion/match_cost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/criterion/match_cost.py -------------------------------------------------------------------------------- /propvg/core/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/layers/__init__.py -------------------------------------------------------------------------------- /propvg/core/layers/mdetr_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/layers/mdetr_transformer.py -------------------------------------------------------------------------------- /propvg/core/layers/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/layers/transformer.py -------------------------------------------------------------------------------- /propvg/core/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/losses/__init__.py -------------------------------------------------------------------------------- /propvg/core/losses/label_smooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/losses/label_smooth.py -------------------------------------------------------------------------------- /propvg/core/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/matcher.py -------------------------------------------------------------------------------- /propvg/core/matcher/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/matcher/__init__.py -------------------------------------------------------------------------------- /propvg/core/matcher/match_cost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/matcher/match_cost.py -------------------------------------------------------------------------------- /propvg/core/matcher/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/matcher/matcher.py -------------------------------------------------------------------------------- /propvg/core/matcher/modified_matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/matcher/modified_matcher.py -------------------------------------------------------------------------------- /propvg/core/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/optimizer.py -------------------------------------------------------------------------------- /propvg/core/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/scheduler.py -------------------------------------------------------------------------------- /propvg/core/structure/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/structure/__init__.py -------------------------------------------------------------------------------- /propvg/core/structure/boxes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/structure/boxes.py -------------------------------------------------------------------------------- /propvg/core/structure/instances.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/structure/instances.py -------------------------------------------------------------------------------- /propvg/core/structure/postprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/structure/postprocessing.py -------------------------------------------------------------------------------- /propvg/core/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/core/utils.py -------------------------------------------------------------------------------- /propvg/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/datasets/__init__.py -------------------------------------------------------------------------------- /propvg/datasets/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/datasets/base.py -------------------------------------------------------------------------------- /propvg/datasets/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/datasets/builder.py -------------------------------------------------------------------------------- /propvg/datasets/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/datasets/pipelines/__init__.py -------------------------------------------------------------------------------- /propvg/datasets/pipelines/compose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/datasets/pipelines/compose.py -------------------------------------------------------------------------------- /propvg/datasets/pipelines/formatting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/datasets/pipelines/formatting.py -------------------------------------------------------------------------------- /propvg/datasets/pipelines/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/datasets/pipelines/loading.py -------------------------------------------------------------------------------- /propvg/datasets/pipelines/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/datasets/pipelines/mask.py -------------------------------------------------------------------------------- /propvg/datasets/pipelines/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/datasets/pipelines/transforms.py -------------------------------------------------------------------------------- /propvg/datasets/pipelines/vgtr_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/datasets/pipelines/vgtr_aug.py -------------------------------------------------------------------------------- /propvg/datasets/read.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/datasets/read.py -------------------------------------------------------------------------------- /propvg/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/datasets/utils.py -------------------------------------------------------------------------------- /propvg/datasets/vgtr_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/datasets/vgtr_dataset.py -------------------------------------------------------------------------------- /propvg/evaluations/grefcoco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/evaluations/grefcoco.py -------------------------------------------------------------------------------- /propvg/evaluations/refzom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/evaluations/refzom.py -------------------------------------------------------------------------------- /propvg/evaluations/rrefcoco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/evaluations/rrefcoco.py -------------------------------------------------------------------------------- /propvg/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/layers/__init__.py -------------------------------------------------------------------------------- /propvg/layers/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/layers/attention.py -------------------------------------------------------------------------------- /propvg/layers/box_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/layers/box_ops.py -------------------------------------------------------------------------------- /propvg/layers/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/layers/conv.py -------------------------------------------------------------------------------- /propvg/layers/dcn_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/layers/dcn_v3.py -------------------------------------------------------------------------------- /propvg/layers/denoising.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/layers/denoising.py -------------------------------------------------------------------------------- /propvg/layers/layer_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/layers/layer_norm.py -------------------------------------------------------------------------------- /propvg/layers/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/layers/mlp.py -------------------------------------------------------------------------------- /propvg/layers/multi_scale_deform_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/layers/multi_scale_deform_attn.py -------------------------------------------------------------------------------- /propvg/layers/position_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/layers/position_embedding.py -------------------------------------------------------------------------------- /propvg/layers/shape_spec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/layers/shape_spec.py -------------------------------------------------------------------------------- /propvg/layers/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/layers/transformer.py -------------------------------------------------------------------------------- /propvg/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/__init__.py -------------------------------------------------------------------------------- /propvg/models/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/builder.py -------------------------------------------------------------------------------- /propvg/models/det_seg/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/det_seg/__init__.py -------------------------------------------------------------------------------- /propvg/models/det_seg/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/det_seg/base.py -------------------------------------------------------------------------------- /propvg/models/det_seg/mix_gref_unimodel_OMG.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/det_seg/mix_gref_unimodel_OMG.py -------------------------------------------------------------------------------- /propvg/models/det_seg/mix_ref_unimodel_OMG.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/det_seg/mix_ref_unimodel_OMG.py -------------------------------------------------------------------------------- /propvg/models/det_seg/one_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/det_seg/one_stage.py -------------------------------------------------------------------------------- /propvg/models/fusions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/fusions/__init__.py -------------------------------------------------------------------------------- /propvg/models/fusions/fusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/fusions/fusion.py -------------------------------------------------------------------------------- /propvg/models/fusions/mdetr_fusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/fusions/mdetr_fusion.py -------------------------------------------------------------------------------- /propvg/models/fusions/simple_fpn_fusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/fusions/simple_fpn_fusion.py -------------------------------------------------------------------------------- /propvg/models/heads/GTMHead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/heads/GTMHead.py -------------------------------------------------------------------------------- /propvg/models/heads/REFHead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/heads/REFHead.py -------------------------------------------------------------------------------- /propvg/models/heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/heads/__init__.py -------------------------------------------------------------------------------- /propvg/models/heads/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/heads/modules.py -------------------------------------------------------------------------------- /propvg/models/heads/projection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/heads/projection.py -------------------------------------------------------------------------------- /propvg/models/heads/tgqs_kd_detr_head/tgqs_kd_detr_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/heads/tgqs_kd_detr_head/tgqs_kd_detr_head.py -------------------------------------------------------------------------------- /propvg/models/heads/tgqs_kd_detr_head/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/heads/tgqs_kd_detr_head/transformer.py -------------------------------------------------------------------------------- /propvg/models/heads/tgqs_kd_detr_head/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/heads/tgqs_kd_detr_head/utils.py -------------------------------------------------------------------------------- /propvg/models/heads/transformers/baselayer_transformer.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /propvg/models/heads/transformers/deformable_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/heads/transformers/deformable_transformer.py -------------------------------------------------------------------------------- /propvg/models/heads/transformers/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/heads/transformers/transformer.py -------------------------------------------------------------------------------- /propvg/models/heads/transformers/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/heads/transformers/utils.py -------------------------------------------------------------------------------- /propvg/models/heads/unet_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/heads/unet_head.py -------------------------------------------------------------------------------- /propvg/models/lan_encs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/lan_encs/__init__.py -------------------------------------------------------------------------------- /propvg/models/lan_encs/bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/lan_encs/bert.py -------------------------------------------------------------------------------- /propvg/models/lan_encs/lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/lan_encs/lstm.py -------------------------------------------------------------------------------- /propvg/models/lan_encs/rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/lan_encs/rnn.py -------------------------------------------------------------------------------- /propvg/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/losses/__init__.py -------------------------------------------------------------------------------- /propvg/models/losses/boxloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/losses/boxloss.py -------------------------------------------------------------------------------- /propvg/models/losses/cem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/losses/cem.py -------------------------------------------------------------------------------- /propvg/models/losses/clip_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/losses/clip_loss.py -------------------------------------------------------------------------------- /propvg/models/losses/contristiveloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/losses/contristiveloss.py -------------------------------------------------------------------------------- /propvg/models/losses/segloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/losses/segloss.py -------------------------------------------------------------------------------- /propvg/models/postprocess/nls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/postprocess/nls.py -------------------------------------------------------------------------------- /propvg/models/postprocess/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/postprocess/nms.py -------------------------------------------------------------------------------- /propvg/models/postprocess/nms_reverse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/postprocess/nms_reverse.py -------------------------------------------------------------------------------- /propvg/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/utils.py -------------------------------------------------------------------------------- /propvg/models/vis_encs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/vis_encs/__init__.py -------------------------------------------------------------------------------- /propvg/models/vis_encs/beit/beit3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/vis_encs/beit/beit3.py -------------------------------------------------------------------------------- /propvg/models/vis_encs/beit/beit3_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/vis_encs/beit/beit3_base.py -------------------------------------------------------------------------------- /propvg/models/vis_encs/beit/modeling_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/vis_encs/beit/modeling_finetune.py -------------------------------------------------------------------------------- /propvg/models/vis_encs/beit/modeling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/vis_encs/beit/modeling_utils.py -------------------------------------------------------------------------------- /propvg/models/vis_encs/beit/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/vis_encs/beit/utils.py -------------------------------------------------------------------------------- /propvg/models/vis_encs/cspdarknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/vis_encs/cspdarknet.py -------------------------------------------------------------------------------- /propvg/models/vis_encs/darknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/vis_encs/darknet.py -------------------------------------------------------------------------------- /propvg/models/vis_encs/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/models/vis_encs/utils.py -------------------------------------------------------------------------------- /propvg/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/utils/__init__.py -------------------------------------------------------------------------------- /propvg/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/utils/checkpoint.py -------------------------------------------------------------------------------- /propvg/utils/det_cam_visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/utils/det_cam_visualizer.py -------------------------------------------------------------------------------- /propvg/utils/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/utils/dist.py -------------------------------------------------------------------------------- /propvg/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/utils/distributed.py -------------------------------------------------------------------------------- /propvg/utils/gradcam_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/utils/gradcam_models.py -------------------------------------------------------------------------------- /propvg/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/utils/logger.py -------------------------------------------------------------------------------- /propvg/utils/visual_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/utils/visual_utils.py -------------------------------------------------------------------------------- /propvg/utils/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/utils/visualizer.py -------------------------------------------------------------------------------- /propvg/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/propvg/version.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/setup.py -------------------------------------------------------------------------------- /tools/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/tools/demo.py -------------------------------------------------------------------------------- /tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/tools/dist_test.sh -------------------------------------------------------------------------------- /tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/tools/dist_train.sh -------------------------------------------------------------------------------- /tools/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/tools/inference.py -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/PropVG/HEAD/tools/train.py --------------------------------------------------------------------------------