├── .gitignore ├── LICENSE ├── README.md ├── configs ├── Base-CRCNN-COCO.yaml ├── O365_CRFR50_CAS_2x.yaml ├── OID_CRFR50_CAS_2x.yaml ├── Partitioned_COIM_R50_6x+2x.yaml ├── Partitioned_COI_R50_2x.yaml ├── Partitioned_COI_R50_6x.yaml ├── Partitioned_COI_R50_8x.yaml ├── Partitioned_COI_RS101_2x.yaml ├── Unified_human_OCI_R50_2x.yaml ├── Unified_learned_OCIM_R50_6x+2x.yaml ├── Unified_learned_OCIM_RS200_6x+2x.yaml ├── Unified_learned_OCI_R50_2x.yaml ├── Unified_learned_OCI_R50_6x.yaml ├── Unified_learned_OCI_R50_8x.yaml └── Unified_learned_OCI_RS200_6x.yaml ├── datasets ├── README.md ├── label_spaces │ ├── leaned_mAP_tau30_668.csv │ ├── leaned_mAP_tau30_668.json │ ├── learned_mAP+M.csv │ ├── learned_mAP+M.json │ ├── learned_mAP+M_labelmap_test.json │ ├── learned_mAP.csv │ ├── learned_mAP.json │ ├── manual.csv │ └── manual.json ├── prepare_ade20k_sem_seg.py ├── prepare_cocofied_lvis.py ├── prepare_for_tests.sh └── prepare_panoptic_fpn.py ├── demo.py ├── docs ├── DATASETS.md ├── REPRODUCE.md ├── example_output.jpg ├── example_output2.jpg └── unidet_teaser.jpg ├── images ├── 16004479832_a748d55f21_k.jpg ├── 17790319373_bd19b24cfc_k.jpg ├── 18124840932_e42b3e377c_k.jpg ├── 19064748793_bb942deea1_k.jpg ├── 24274813513_0cfd2ce6d0_k.jpg ├── 33823288584_1d21cf0a26_k.jpg ├── 33887522274_eebd074106_k.jpg ├── 34501842524_3c858b3080_k.jpg ├── NOTICE ├── au0031_100000.jpg ├── bh0002_100000.jpg ├── bt0023_100000.jpg └── cn0000_100000.jpg ├── requirements.txt ├── tools ├── UniDet_learn_labelspace_mAP.ipynb ├── convert_datasets │ ├── convert_crowdhuman.py │ ├── convert_mappilary.py │ ├── convert_oid.py │ └── get_oid_hierarchy.py ├── create_unified_label_json.py ├── create_unified_label_json_4datasets.py ├── dump_categories.py ├── dump_outputs.py └── match_test_datasets.py ├── train_net.py └── unidet ├── __init__.py ├── config.py ├── data ├── custom_dataset_dataloader.py ├── datasets │ ├── cityscapes_cocoformat.py │ ├── crowdhuman.py │ ├── det_categories.py │ ├── inst_categories.py │ ├── kitti.py │ ├── mapillary.py │ ├── objects365.py │ ├── oid.py │ ├── register_oid.py │ ├── scannet.py │ ├── viper.py │ ├── voc_cocoformat.py │ └── wilddash.py └── multi_dataset_dataloader.py ├── evaluation ├── multi_dataset_evaluator.py └── oideval.py ├── modeling ├── backbone │ ├── fpn_p5.py │ ├── resnest.py │ └── splat.py ├── meta_arch │ ├── split_rcnn.py │ └── unified_rcnn.py └── roi_heads │ ├── custom_fast_rcnn.py │ ├── custom_roi_heads.py │ ├── multi_dataset_fast_rcnn.py │ ├── split_roi_heads.py │ └── unified_roi_heads.py └── predictor.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/README.md -------------------------------------------------------------------------------- /configs/Base-CRCNN-COCO.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/configs/Base-CRCNN-COCO.yaml -------------------------------------------------------------------------------- /configs/O365_CRFR50_CAS_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/configs/O365_CRFR50_CAS_2x.yaml -------------------------------------------------------------------------------- /configs/OID_CRFR50_CAS_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/configs/OID_CRFR50_CAS_2x.yaml -------------------------------------------------------------------------------- /configs/Partitioned_COIM_R50_6x+2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/configs/Partitioned_COIM_R50_6x+2x.yaml -------------------------------------------------------------------------------- /configs/Partitioned_COI_R50_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/configs/Partitioned_COI_R50_2x.yaml -------------------------------------------------------------------------------- /configs/Partitioned_COI_R50_6x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/configs/Partitioned_COI_R50_6x.yaml -------------------------------------------------------------------------------- /configs/Partitioned_COI_R50_8x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/configs/Partitioned_COI_R50_8x.yaml -------------------------------------------------------------------------------- /configs/Partitioned_COI_RS101_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/configs/Partitioned_COI_RS101_2x.yaml -------------------------------------------------------------------------------- /configs/Unified_human_OCI_R50_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/configs/Unified_human_OCI_R50_2x.yaml -------------------------------------------------------------------------------- /configs/Unified_learned_OCIM_R50_6x+2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/configs/Unified_learned_OCIM_R50_6x+2x.yaml -------------------------------------------------------------------------------- /configs/Unified_learned_OCIM_RS200_6x+2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/configs/Unified_learned_OCIM_RS200_6x+2x.yaml -------------------------------------------------------------------------------- /configs/Unified_learned_OCI_R50_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/configs/Unified_learned_OCI_R50_2x.yaml -------------------------------------------------------------------------------- /configs/Unified_learned_OCI_R50_6x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/configs/Unified_learned_OCI_R50_6x.yaml -------------------------------------------------------------------------------- /configs/Unified_learned_OCI_R50_8x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/configs/Unified_learned_OCI_R50_8x.yaml -------------------------------------------------------------------------------- /configs/Unified_learned_OCI_RS200_6x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/configs/Unified_learned_OCI_RS200_6x.yaml -------------------------------------------------------------------------------- /datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/datasets/README.md -------------------------------------------------------------------------------- /datasets/label_spaces/leaned_mAP_tau30_668.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/datasets/label_spaces/leaned_mAP_tau30_668.csv -------------------------------------------------------------------------------- /datasets/label_spaces/leaned_mAP_tau30_668.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/datasets/label_spaces/leaned_mAP_tau30_668.json -------------------------------------------------------------------------------- /datasets/label_spaces/learned_mAP+M.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/datasets/label_spaces/learned_mAP+M.csv -------------------------------------------------------------------------------- /datasets/label_spaces/learned_mAP+M.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/datasets/label_spaces/learned_mAP+M.json -------------------------------------------------------------------------------- /datasets/label_spaces/learned_mAP+M_labelmap_test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/datasets/label_spaces/learned_mAP+M_labelmap_test.json -------------------------------------------------------------------------------- /datasets/label_spaces/learned_mAP.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/datasets/label_spaces/learned_mAP.csv -------------------------------------------------------------------------------- /datasets/label_spaces/learned_mAP.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/datasets/label_spaces/learned_mAP.json -------------------------------------------------------------------------------- /datasets/label_spaces/manual.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/datasets/label_spaces/manual.csv -------------------------------------------------------------------------------- /datasets/label_spaces/manual.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/datasets/label_spaces/manual.json -------------------------------------------------------------------------------- /datasets/prepare_ade20k_sem_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/datasets/prepare_ade20k_sem_seg.py -------------------------------------------------------------------------------- /datasets/prepare_cocofied_lvis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/datasets/prepare_cocofied_lvis.py -------------------------------------------------------------------------------- /datasets/prepare_for_tests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/datasets/prepare_for_tests.sh -------------------------------------------------------------------------------- /datasets/prepare_panoptic_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/datasets/prepare_panoptic_fpn.py -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/demo.py -------------------------------------------------------------------------------- /docs/DATASETS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/docs/DATASETS.md -------------------------------------------------------------------------------- /docs/REPRODUCE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/docs/REPRODUCE.md -------------------------------------------------------------------------------- /docs/example_output.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/docs/example_output.jpg -------------------------------------------------------------------------------- /docs/example_output2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/docs/example_output2.jpg -------------------------------------------------------------------------------- /docs/unidet_teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/docs/unidet_teaser.jpg -------------------------------------------------------------------------------- /images/16004479832_a748d55f21_k.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/images/16004479832_a748d55f21_k.jpg -------------------------------------------------------------------------------- /images/17790319373_bd19b24cfc_k.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/images/17790319373_bd19b24cfc_k.jpg -------------------------------------------------------------------------------- /images/18124840932_e42b3e377c_k.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/images/18124840932_e42b3e377c_k.jpg -------------------------------------------------------------------------------- /images/19064748793_bb942deea1_k.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/images/19064748793_bb942deea1_k.jpg -------------------------------------------------------------------------------- /images/24274813513_0cfd2ce6d0_k.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/images/24274813513_0cfd2ce6d0_k.jpg -------------------------------------------------------------------------------- /images/33823288584_1d21cf0a26_k.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/images/33823288584_1d21cf0a26_k.jpg -------------------------------------------------------------------------------- /images/33887522274_eebd074106_k.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/images/33887522274_eebd074106_k.jpg -------------------------------------------------------------------------------- /images/34501842524_3c858b3080_k.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/images/34501842524_3c858b3080_k.jpg -------------------------------------------------------------------------------- /images/NOTICE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/images/NOTICE -------------------------------------------------------------------------------- /images/au0031_100000.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/images/au0031_100000.jpg -------------------------------------------------------------------------------- /images/bh0002_100000.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/images/bh0002_100000.jpg -------------------------------------------------------------------------------- /images/bt0023_100000.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/images/bt0023_100000.jpg -------------------------------------------------------------------------------- /images/cn0000_100000.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/images/cn0000_100000.jpg -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | opencv-python -------------------------------------------------------------------------------- /tools/UniDet_learn_labelspace_mAP.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/tools/UniDet_learn_labelspace_mAP.ipynb -------------------------------------------------------------------------------- /tools/convert_datasets/convert_crowdhuman.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/tools/convert_datasets/convert_crowdhuman.py -------------------------------------------------------------------------------- /tools/convert_datasets/convert_mappilary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/tools/convert_datasets/convert_mappilary.py -------------------------------------------------------------------------------- /tools/convert_datasets/convert_oid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/tools/convert_datasets/convert_oid.py -------------------------------------------------------------------------------- /tools/convert_datasets/get_oid_hierarchy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/tools/convert_datasets/get_oid_hierarchy.py -------------------------------------------------------------------------------- /tools/create_unified_label_json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/tools/create_unified_label_json.py -------------------------------------------------------------------------------- /tools/create_unified_label_json_4datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/tools/create_unified_label_json_4datasets.py -------------------------------------------------------------------------------- /tools/dump_categories.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/tools/dump_categories.py -------------------------------------------------------------------------------- /tools/dump_outputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/tools/dump_outputs.py -------------------------------------------------------------------------------- /tools/match_test_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/tools/match_test_datasets.py -------------------------------------------------------------------------------- /train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/train_net.py -------------------------------------------------------------------------------- /unidet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/__init__.py -------------------------------------------------------------------------------- /unidet/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/config.py -------------------------------------------------------------------------------- /unidet/data/custom_dataset_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/data/custom_dataset_dataloader.py -------------------------------------------------------------------------------- /unidet/data/datasets/cityscapes_cocoformat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/data/datasets/cityscapes_cocoformat.py -------------------------------------------------------------------------------- /unidet/data/datasets/crowdhuman.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/data/datasets/crowdhuman.py -------------------------------------------------------------------------------- /unidet/data/datasets/det_categories.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/data/datasets/det_categories.py -------------------------------------------------------------------------------- /unidet/data/datasets/inst_categories.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/data/datasets/inst_categories.py -------------------------------------------------------------------------------- /unidet/data/datasets/kitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/data/datasets/kitti.py -------------------------------------------------------------------------------- /unidet/data/datasets/mapillary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/data/datasets/mapillary.py -------------------------------------------------------------------------------- /unidet/data/datasets/objects365.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/data/datasets/objects365.py -------------------------------------------------------------------------------- /unidet/data/datasets/oid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/data/datasets/oid.py -------------------------------------------------------------------------------- /unidet/data/datasets/register_oid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/data/datasets/register_oid.py -------------------------------------------------------------------------------- /unidet/data/datasets/scannet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/data/datasets/scannet.py -------------------------------------------------------------------------------- /unidet/data/datasets/viper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/data/datasets/viper.py -------------------------------------------------------------------------------- /unidet/data/datasets/voc_cocoformat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/data/datasets/voc_cocoformat.py -------------------------------------------------------------------------------- /unidet/data/datasets/wilddash.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/data/datasets/wilddash.py -------------------------------------------------------------------------------- /unidet/data/multi_dataset_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/data/multi_dataset_dataloader.py -------------------------------------------------------------------------------- /unidet/evaluation/multi_dataset_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/evaluation/multi_dataset_evaluator.py -------------------------------------------------------------------------------- /unidet/evaluation/oideval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/evaluation/oideval.py -------------------------------------------------------------------------------- /unidet/modeling/backbone/fpn_p5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/modeling/backbone/fpn_p5.py -------------------------------------------------------------------------------- /unidet/modeling/backbone/resnest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/modeling/backbone/resnest.py -------------------------------------------------------------------------------- /unidet/modeling/backbone/splat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/modeling/backbone/splat.py -------------------------------------------------------------------------------- /unidet/modeling/meta_arch/split_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/modeling/meta_arch/split_rcnn.py -------------------------------------------------------------------------------- /unidet/modeling/meta_arch/unified_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/modeling/meta_arch/unified_rcnn.py -------------------------------------------------------------------------------- /unidet/modeling/roi_heads/custom_fast_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/modeling/roi_heads/custom_fast_rcnn.py -------------------------------------------------------------------------------- /unidet/modeling/roi_heads/custom_roi_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/modeling/roi_heads/custom_roi_heads.py -------------------------------------------------------------------------------- /unidet/modeling/roi_heads/multi_dataset_fast_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/modeling/roi_heads/multi_dataset_fast_rcnn.py -------------------------------------------------------------------------------- /unidet/modeling/roi_heads/split_roi_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/modeling/roi_heads/split_roi_heads.py -------------------------------------------------------------------------------- /unidet/modeling/roi_heads/unified_roi_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/modeling/roi_heads/unified_roi_heads.py -------------------------------------------------------------------------------- /unidet/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/UniDet/HEAD/unidet/predictor.py --------------------------------------------------------------------------------