├── .gitignore ├── LICENSE ├── README.md ├── assets └── model.png ├── configs ├── Base_OVAD_C4_1x.yaml ├── Base_OVCOCO_C4_1x.yaml └── BoxAnn_OVAD_C4_1x.yaml ├── ovadb ├── __init__.py ├── config.py ├── custom_solver.py ├── data │ ├── custom_build_augmentation.py │ ├── custom_dataset_dataloader.py │ ├── custom_dataset_mapper.py │ ├── data_utils.py │ └── datasets │ │ ├── coco_ovd.py │ │ ├── ovad.py │ │ ├── register_data.py │ │ └── utils.py ├── evaluation │ ├── ann_box_evaluator.py │ ├── attribute_evaluation.py │ └── custom_coco_eval.py ├── modeling │ ├── debug.py │ ├── logged_module.py │ ├── meta_arch │ │ ├── custom_rcnn.py │ │ └── phrase_custom_rcnn.py │ ├── roi_heads │ │ ├── attribute_heads │ │ │ ├── __init__.py │ │ │ └── attribute_classifier.py │ │ ├── box_heads │ │ │ ├── __init__.py │ │ │ ├── detic_fast_rcnn.py │ │ │ ├── emb_fast_rcnn.py │ │ │ ├── phrase_fast_rcnn.py │ │ │ ├── vlplm_fast_rcnn.py │ │ │ └── zero_shot_classifier.py │ │ ├── res5_roi_heads.py │ │ ├── roi_emb_heads.py │ │ └── vlplm_roi_heads.py │ ├── text │ │ ├── text_encoder.py │ │ └── transformer_encoder.py │ └── utils.py └── utils │ ├── checkpoint.py │ ├── events.py │ └── visualizer.py ├── ovamc ├── data_loader.py ├── misc.py ├── ova_albef.py ├── ova_blip.py ├── ova_blip2.py ├── ova_clip.py ├── ova_open_clip.py └── ova_xvlm.py ├── requirements.txt ├── tools ├── dump_attribute_features.py ├── extract_obj_boxes.py └── make_ovd_json.py └── train_att_net.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/README.md -------------------------------------------------------------------------------- /assets/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/assets/model.png -------------------------------------------------------------------------------- /configs/Base_OVAD_C4_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/configs/Base_OVAD_C4_1x.yaml -------------------------------------------------------------------------------- /configs/Base_OVCOCO_C4_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/configs/Base_OVCOCO_C4_1x.yaml -------------------------------------------------------------------------------- /configs/BoxAnn_OVAD_C4_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/configs/BoxAnn_OVAD_C4_1x.yaml -------------------------------------------------------------------------------- /ovadb/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/__init__.py -------------------------------------------------------------------------------- /ovadb/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/config.py -------------------------------------------------------------------------------- /ovadb/custom_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/custom_solver.py -------------------------------------------------------------------------------- /ovadb/data/custom_build_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/data/custom_build_augmentation.py -------------------------------------------------------------------------------- /ovadb/data/custom_dataset_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/data/custom_dataset_dataloader.py -------------------------------------------------------------------------------- /ovadb/data/custom_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/data/custom_dataset_mapper.py -------------------------------------------------------------------------------- /ovadb/data/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/data/data_utils.py -------------------------------------------------------------------------------- /ovadb/data/datasets/coco_ovd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/data/datasets/coco_ovd.py -------------------------------------------------------------------------------- /ovadb/data/datasets/ovad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/data/datasets/ovad.py -------------------------------------------------------------------------------- /ovadb/data/datasets/register_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/data/datasets/register_data.py -------------------------------------------------------------------------------- /ovadb/data/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/data/datasets/utils.py -------------------------------------------------------------------------------- /ovadb/evaluation/ann_box_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/evaluation/ann_box_evaluator.py -------------------------------------------------------------------------------- /ovadb/evaluation/attribute_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/evaluation/attribute_evaluation.py -------------------------------------------------------------------------------- /ovadb/evaluation/custom_coco_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/evaluation/custom_coco_eval.py -------------------------------------------------------------------------------- /ovadb/modeling/debug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/modeling/debug.py -------------------------------------------------------------------------------- /ovadb/modeling/logged_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/modeling/logged_module.py -------------------------------------------------------------------------------- /ovadb/modeling/meta_arch/custom_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/modeling/meta_arch/custom_rcnn.py -------------------------------------------------------------------------------- /ovadb/modeling/meta_arch/phrase_custom_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/modeling/meta_arch/phrase_custom_rcnn.py -------------------------------------------------------------------------------- /ovadb/modeling/roi_heads/attribute_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/modeling/roi_heads/attribute_heads/__init__.py -------------------------------------------------------------------------------- /ovadb/modeling/roi_heads/attribute_heads/attribute_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/modeling/roi_heads/attribute_heads/attribute_classifier.py -------------------------------------------------------------------------------- /ovadb/modeling/roi_heads/box_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/modeling/roi_heads/box_heads/__init__.py -------------------------------------------------------------------------------- /ovadb/modeling/roi_heads/box_heads/detic_fast_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/modeling/roi_heads/box_heads/detic_fast_rcnn.py -------------------------------------------------------------------------------- /ovadb/modeling/roi_heads/box_heads/emb_fast_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/modeling/roi_heads/box_heads/emb_fast_rcnn.py -------------------------------------------------------------------------------- /ovadb/modeling/roi_heads/box_heads/phrase_fast_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/modeling/roi_heads/box_heads/phrase_fast_rcnn.py -------------------------------------------------------------------------------- /ovadb/modeling/roi_heads/box_heads/vlplm_fast_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/modeling/roi_heads/box_heads/vlplm_fast_rcnn.py -------------------------------------------------------------------------------- /ovadb/modeling/roi_heads/box_heads/zero_shot_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/modeling/roi_heads/box_heads/zero_shot_classifier.py -------------------------------------------------------------------------------- /ovadb/modeling/roi_heads/res5_roi_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/modeling/roi_heads/res5_roi_heads.py -------------------------------------------------------------------------------- /ovadb/modeling/roi_heads/roi_emb_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/modeling/roi_heads/roi_emb_heads.py -------------------------------------------------------------------------------- /ovadb/modeling/roi_heads/vlplm_roi_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/modeling/roi_heads/vlplm_roi_heads.py -------------------------------------------------------------------------------- /ovadb/modeling/text/text_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/modeling/text/text_encoder.py -------------------------------------------------------------------------------- /ovadb/modeling/text/transformer_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/modeling/text/transformer_encoder.py -------------------------------------------------------------------------------- /ovadb/modeling/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/modeling/utils.py -------------------------------------------------------------------------------- /ovadb/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/utils/checkpoint.py -------------------------------------------------------------------------------- /ovadb/utils/events.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/utils/events.py -------------------------------------------------------------------------------- /ovadb/utils/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovadb/utils/visualizer.py -------------------------------------------------------------------------------- /ovamc/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovamc/data_loader.py -------------------------------------------------------------------------------- /ovamc/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovamc/misc.py -------------------------------------------------------------------------------- /ovamc/ova_albef.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovamc/ova_albef.py -------------------------------------------------------------------------------- /ovamc/ova_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovamc/ova_blip.py -------------------------------------------------------------------------------- /ovamc/ova_blip2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovamc/ova_blip2.py -------------------------------------------------------------------------------- /ovamc/ova_clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovamc/ova_clip.py -------------------------------------------------------------------------------- /ovamc/ova_open_clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovamc/ova_open_clip.py -------------------------------------------------------------------------------- /ovamc/ova_xvlm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/ovamc/ova_xvlm.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/requirements.txt -------------------------------------------------------------------------------- /tools/dump_attribute_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/tools/dump_attribute_features.py -------------------------------------------------------------------------------- /tools/extract_obj_boxes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/tools/extract_obj_boxes.py -------------------------------------------------------------------------------- /tools/make_ovd_json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/tools/make_ovd_json.py -------------------------------------------------------------------------------- /train_att_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OVAD-Benchmark/ovad-benchmark-code/HEAD/train_att_net.py --------------------------------------------------------------------------------