├── .gitignore ├── .gitmodules ├── LICENSE ├── README.md ├── gpv2 ├── __init__.py ├── build_image_features │ ├── compute_features.py │ ├── precompute_dataset_features.py │ └── precompute_image_features.py ├── data │ ├── coco_categories.json │ ├── dataset.py │ ├── dce_dataset.py │ ├── gpv_datasets.py │ ├── synonyms.py │ ├── webqa_adj_types.json │ ├── webqa_dataset.py │ └── webqa_templates.py ├── detr │ ├── LICENSE │ ├── box_ops.py │ ├── detr_misc.py │ ├── matcher.py │ └── set_criterion.py ├── download_data.py ├── eval │ ├── build_submission_files.py │ ├── compute_topn_predictions.py │ ├── dataset_cli.py │ ├── eval_model.py │ ├── evaluation.py │ ├── run_on_image_id.py │ └── vqa_eval.py ├── experiments │ ├── train_gpv2.py │ └── trainer_cli.py ├── file_paths.py ├── image_featurizer │ ├── __init__.py │ ├── image_featurizer.py │ ├── precomputed_features.py │ └── vinvl_featurizer.py ├── model │ ├── __init__.py │ ├── allennlp_beamsearch.py │ ├── collate.py │ ├── gpv2.py │ ├── gpv_example.py │ ├── layers.py │ ├── load_model.py │ ├── loss.py │ ├── model.py │ ├── preprocess_example.py │ └── t5_custom.py ├── train │ ├── evaluator.py │ ├── fixed_adamw.py │ ├── optimizer.py │ ├── runner.py │ ├── samplers.py │ ├── stratified_subset_sampler.py │ ├── trainer.py │ └── vqa2_eval_data.py ├── utils │ ├── downloader.py │ ├── image_utils.py │ ├── py_utils.py │ ├── pytorch_utils.py │ ├── quiet_ptbtokenizer.py │ └── to_params.py └── vinvl │ ├── LICENSE │ ├── __init__.py │ ├── attr_rcnn.py │ ├── attribute_head │ ├── __init__.py │ ├── attribute_head.py │ ├── inference.py │ ├── loss.py │ ├── roi_attribute_feature_extractors.py │ └── roi_attribute_predictors.py │ ├── backbone │ ├── __init__.py │ ├── backbone.py │ ├── fpn.py │ └── resnet.py │ ├── balanced_positive_negative_sampler.py │ ├── box_coder.py │ ├── defaults.py │ ├── detector │ ├── __init__.py │ ├── detectors.py │ └── generalized_rcnn.py │ ├── get_vinvl.py │ ├── layers │ ├── __init__.py │ ├── _utils.py │ ├── batch_norm.py │ ├── misc.py │ ├── nms.py │ ├── roi_align.py │ ├── roi_pool.py │ ├── sigmoid_focal_loss.py │ └── smooth_l1_loss.py │ ├── make_layers.py │ ├── matcher.py │ ├── modelling_utils.py │ ├── poolers.py │ ├── registry.py │ ├── roi_heads │ ├── __init__.py │ ├── box_head │ │ ├── __init__.py │ │ ├── box_head.py │ │ ├── inference.py │ │ ├── loss.py │ │ ├── roi_box_feature_extractors.py │ │ └── roi_box_predictors.py │ ├── mask_head │ │ ├── __init__.py │ │ ├── inference.py │ │ ├── loss.py │ │ ├── mask_head.py │ │ ├── roi_mask_feature_extractors.py │ │ └── roi_mask_predictors.py │ └── roi_heads.py │ ├── rpn │ ├── __init__.py │ ├── anchor_generator.py │ ├── inference.py │ ├── loss.py │ ├── rpn.py │ └── utils.py │ ├── sg_defaults.py │ ├── structures │ ├── __init__.py │ ├── bounding_box.py │ ├── bounding_box_pair.py │ ├── boxlist_ops.py │ └── image_list.py │ ├── transforms.py │ ├── utils │ ├── README.md │ ├── __init__.py │ ├── c2_model_loading.py │ ├── checkpoint.py │ ├── comm.py │ ├── cv2_util.py │ ├── dataset_utils.py │ ├── imports.py │ ├── miscellaneous.py │ ├── model_serialization.py │ ├── model_zoo.py │ └── registry.py │ └── vinvl_x152c4.yaml └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/.gitmodules -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/README.md -------------------------------------------------------------------------------- /gpv2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gpv2/build_image_features/compute_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/build_image_features/compute_features.py -------------------------------------------------------------------------------- /gpv2/build_image_features/precompute_dataset_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/build_image_features/precompute_dataset_features.py -------------------------------------------------------------------------------- /gpv2/build_image_features/precompute_image_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/build_image_features/precompute_image_features.py -------------------------------------------------------------------------------- /gpv2/data/coco_categories.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/data/coco_categories.json -------------------------------------------------------------------------------- /gpv2/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/data/dataset.py -------------------------------------------------------------------------------- /gpv2/data/dce_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/data/dce_dataset.py -------------------------------------------------------------------------------- /gpv2/data/gpv_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/data/gpv_datasets.py -------------------------------------------------------------------------------- /gpv2/data/synonyms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/data/synonyms.py -------------------------------------------------------------------------------- /gpv2/data/webqa_adj_types.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/data/webqa_adj_types.json -------------------------------------------------------------------------------- /gpv2/data/webqa_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/data/webqa_dataset.py -------------------------------------------------------------------------------- /gpv2/data/webqa_templates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/data/webqa_templates.py -------------------------------------------------------------------------------- /gpv2/detr/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/detr/LICENSE -------------------------------------------------------------------------------- /gpv2/detr/box_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/detr/box_ops.py -------------------------------------------------------------------------------- /gpv2/detr/detr_misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/detr/detr_misc.py -------------------------------------------------------------------------------- /gpv2/detr/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/detr/matcher.py -------------------------------------------------------------------------------- /gpv2/detr/set_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/detr/set_criterion.py -------------------------------------------------------------------------------- /gpv2/download_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/download_data.py -------------------------------------------------------------------------------- /gpv2/eval/build_submission_files.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/eval/build_submission_files.py -------------------------------------------------------------------------------- /gpv2/eval/compute_topn_predictions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/eval/compute_topn_predictions.py -------------------------------------------------------------------------------- /gpv2/eval/dataset_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/eval/dataset_cli.py -------------------------------------------------------------------------------- /gpv2/eval/eval_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/eval/eval_model.py -------------------------------------------------------------------------------- /gpv2/eval/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/eval/evaluation.py -------------------------------------------------------------------------------- /gpv2/eval/run_on_image_id.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/eval/run_on_image_id.py -------------------------------------------------------------------------------- /gpv2/eval/vqa_eval.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gpv2/experiments/train_gpv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/experiments/train_gpv2.py -------------------------------------------------------------------------------- /gpv2/experiments/trainer_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/experiments/trainer_cli.py -------------------------------------------------------------------------------- /gpv2/file_paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/file_paths.py -------------------------------------------------------------------------------- /gpv2/image_featurizer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gpv2/image_featurizer/image_featurizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/image_featurizer/image_featurizer.py -------------------------------------------------------------------------------- /gpv2/image_featurizer/precomputed_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/image_featurizer/precomputed_features.py -------------------------------------------------------------------------------- /gpv2/image_featurizer/vinvl_featurizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/image_featurizer/vinvl_featurizer.py -------------------------------------------------------------------------------- /gpv2/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gpv2/model/allennlp_beamsearch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/model/allennlp_beamsearch.py -------------------------------------------------------------------------------- /gpv2/model/collate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/model/collate.py -------------------------------------------------------------------------------- /gpv2/model/gpv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/model/gpv2.py -------------------------------------------------------------------------------- /gpv2/model/gpv_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/model/gpv_example.py -------------------------------------------------------------------------------- /gpv2/model/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/model/layers.py -------------------------------------------------------------------------------- /gpv2/model/load_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/model/load_model.py -------------------------------------------------------------------------------- /gpv2/model/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/model/loss.py -------------------------------------------------------------------------------- /gpv2/model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/model/model.py -------------------------------------------------------------------------------- /gpv2/model/preprocess_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/model/preprocess_example.py -------------------------------------------------------------------------------- /gpv2/model/t5_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/model/t5_custom.py -------------------------------------------------------------------------------- /gpv2/train/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/train/evaluator.py -------------------------------------------------------------------------------- /gpv2/train/fixed_adamw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/train/fixed_adamw.py -------------------------------------------------------------------------------- /gpv2/train/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/train/optimizer.py -------------------------------------------------------------------------------- /gpv2/train/runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/train/runner.py -------------------------------------------------------------------------------- /gpv2/train/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/train/samplers.py -------------------------------------------------------------------------------- /gpv2/train/stratified_subset_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/train/stratified_subset_sampler.py -------------------------------------------------------------------------------- /gpv2/train/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/train/trainer.py -------------------------------------------------------------------------------- /gpv2/train/vqa2_eval_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/train/vqa2_eval_data.py -------------------------------------------------------------------------------- /gpv2/utils/downloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/utils/downloader.py -------------------------------------------------------------------------------- /gpv2/utils/image_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/utils/image_utils.py -------------------------------------------------------------------------------- /gpv2/utils/py_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/utils/py_utils.py -------------------------------------------------------------------------------- /gpv2/utils/pytorch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/utils/pytorch_utils.py -------------------------------------------------------------------------------- /gpv2/utils/quiet_ptbtokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/utils/quiet_ptbtokenizer.py -------------------------------------------------------------------------------- /gpv2/utils/to_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/utils/to_params.py -------------------------------------------------------------------------------- /gpv2/vinvl/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/LICENSE -------------------------------------------------------------------------------- /gpv2/vinvl/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gpv2/vinvl/attr_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/attr_rcnn.py -------------------------------------------------------------------------------- /gpv2/vinvl/attribute_head/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/attribute_head/__init__.py -------------------------------------------------------------------------------- /gpv2/vinvl/attribute_head/attribute_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/attribute_head/attribute_head.py -------------------------------------------------------------------------------- /gpv2/vinvl/attribute_head/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/attribute_head/inference.py -------------------------------------------------------------------------------- /gpv2/vinvl/attribute_head/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/attribute_head/loss.py -------------------------------------------------------------------------------- /gpv2/vinvl/attribute_head/roi_attribute_feature_extractors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/attribute_head/roi_attribute_feature_extractors.py -------------------------------------------------------------------------------- /gpv2/vinvl/attribute_head/roi_attribute_predictors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/attribute_head/roi_attribute_predictors.py -------------------------------------------------------------------------------- /gpv2/vinvl/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/backbone/__init__.py -------------------------------------------------------------------------------- /gpv2/vinvl/backbone/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/backbone/backbone.py -------------------------------------------------------------------------------- /gpv2/vinvl/backbone/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/backbone/fpn.py -------------------------------------------------------------------------------- /gpv2/vinvl/backbone/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/backbone/resnet.py -------------------------------------------------------------------------------- /gpv2/vinvl/balanced_positive_negative_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/balanced_positive_negative_sampler.py -------------------------------------------------------------------------------- /gpv2/vinvl/box_coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/box_coder.py -------------------------------------------------------------------------------- /gpv2/vinvl/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/defaults.py -------------------------------------------------------------------------------- /gpv2/vinvl/detector/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/detector/__init__.py -------------------------------------------------------------------------------- /gpv2/vinvl/detector/detectors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/detector/detectors.py -------------------------------------------------------------------------------- /gpv2/vinvl/detector/generalized_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/detector/generalized_rcnn.py -------------------------------------------------------------------------------- /gpv2/vinvl/get_vinvl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/get_vinvl.py -------------------------------------------------------------------------------- /gpv2/vinvl/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/layers/__init__.py -------------------------------------------------------------------------------- /gpv2/vinvl/layers/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/layers/_utils.py -------------------------------------------------------------------------------- /gpv2/vinvl/layers/batch_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/layers/batch_norm.py -------------------------------------------------------------------------------- /gpv2/vinvl/layers/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/layers/misc.py -------------------------------------------------------------------------------- /gpv2/vinvl/layers/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/layers/nms.py -------------------------------------------------------------------------------- /gpv2/vinvl/layers/roi_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/layers/roi_align.py -------------------------------------------------------------------------------- /gpv2/vinvl/layers/roi_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/layers/roi_pool.py -------------------------------------------------------------------------------- /gpv2/vinvl/layers/sigmoid_focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/layers/sigmoid_focal_loss.py -------------------------------------------------------------------------------- /gpv2/vinvl/layers/smooth_l1_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/layers/smooth_l1_loss.py -------------------------------------------------------------------------------- /gpv2/vinvl/make_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/make_layers.py -------------------------------------------------------------------------------- /gpv2/vinvl/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/matcher.py -------------------------------------------------------------------------------- /gpv2/vinvl/modelling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/modelling_utils.py -------------------------------------------------------------------------------- /gpv2/vinvl/poolers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/poolers.py -------------------------------------------------------------------------------- /gpv2/vinvl/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/registry.py -------------------------------------------------------------------------------- /gpv2/vinvl/roi_heads/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gpv2/vinvl/roi_heads/box_head/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gpv2/vinvl/roi_heads/box_head/box_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/roi_heads/box_head/box_head.py -------------------------------------------------------------------------------- /gpv2/vinvl/roi_heads/box_head/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/roi_heads/box_head/inference.py -------------------------------------------------------------------------------- /gpv2/vinvl/roi_heads/box_head/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/roi_heads/box_head/loss.py -------------------------------------------------------------------------------- /gpv2/vinvl/roi_heads/box_head/roi_box_feature_extractors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/roi_heads/box_head/roi_box_feature_extractors.py -------------------------------------------------------------------------------- /gpv2/vinvl/roi_heads/box_head/roi_box_predictors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/roi_heads/box_head/roi_box_predictors.py -------------------------------------------------------------------------------- /gpv2/vinvl/roi_heads/mask_head/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gpv2/vinvl/roi_heads/mask_head/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/roi_heads/mask_head/inference.py -------------------------------------------------------------------------------- /gpv2/vinvl/roi_heads/mask_head/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/roi_heads/mask_head/loss.py -------------------------------------------------------------------------------- /gpv2/vinvl/roi_heads/mask_head/mask_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/roi_heads/mask_head/mask_head.py -------------------------------------------------------------------------------- /gpv2/vinvl/roi_heads/mask_head/roi_mask_feature_extractors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/roi_heads/mask_head/roi_mask_feature_extractors.py -------------------------------------------------------------------------------- /gpv2/vinvl/roi_heads/mask_head/roi_mask_predictors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/roi_heads/mask_head/roi_mask_predictors.py -------------------------------------------------------------------------------- /gpv2/vinvl/roi_heads/roi_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/roi_heads/roi_heads.py -------------------------------------------------------------------------------- /gpv2/vinvl/rpn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/rpn/__init__.py -------------------------------------------------------------------------------- /gpv2/vinvl/rpn/anchor_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/rpn/anchor_generator.py -------------------------------------------------------------------------------- /gpv2/vinvl/rpn/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/rpn/inference.py -------------------------------------------------------------------------------- /gpv2/vinvl/rpn/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/rpn/loss.py -------------------------------------------------------------------------------- /gpv2/vinvl/rpn/rpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/rpn/rpn.py -------------------------------------------------------------------------------- /gpv2/vinvl/rpn/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/rpn/utils.py -------------------------------------------------------------------------------- /gpv2/vinvl/sg_defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/sg_defaults.py -------------------------------------------------------------------------------- /gpv2/vinvl/structures/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gpv2/vinvl/structures/bounding_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/structures/bounding_box.py -------------------------------------------------------------------------------- /gpv2/vinvl/structures/bounding_box_pair.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/structures/bounding_box_pair.py -------------------------------------------------------------------------------- /gpv2/vinvl/structures/boxlist_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/structures/boxlist_ops.py -------------------------------------------------------------------------------- /gpv2/vinvl/structures/image_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/structures/image_list.py -------------------------------------------------------------------------------- /gpv2/vinvl/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/transforms.py -------------------------------------------------------------------------------- /gpv2/vinvl/utils/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/utils/README.md -------------------------------------------------------------------------------- /gpv2/vinvl/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gpv2/vinvl/utils/c2_model_loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/utils/c2_model_loading.py -------------------------------------------------------------------------------- /gpv2/vinvl/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/utils/checkpoint.py -------------------------------------------------------------------------------- /gpv2/vinvl/utils/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/utils/comm.py -------------------------------------------------------------------------------- /gpv2/vinvl/utils/cv2_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/utils/cv2_util.py -------------------------------------------------------------------------------- /gpv2/vinvl/utils/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/utils/dataset_utils.py -------------------------------------------------------------------------------- /gpv2/vinvl/utils/imports.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/utils/imports.py -------------------------------------------------------------------------------- /gpv2/vinvl/utils/miscellaneous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/utils/miscellaneous.py -------------------------------------------------------------------------------- /gpv2/vinvl/utils/model_serialization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/utils/model_serialization.py -------------------------------------------------------------------------------- /gpv2/vinvl/utils/model_zoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/utils/model_zoo.py -------------------------------------------------------------------------------- /gpv2/vinvl/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/utils/registry.py -------------------------------------------------------------------------------- /gpv2/vinvl/vinvl_x152c4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/gpv2/vinvl/vinvl_x152c4.yaml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/gpv2/HEAD/requirements.txt --------------------------------------------------------------------------------