├── .github ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── Detectron2-Logo-Horz.svg ├── ISSUE_TEMPLATE.md ├── ISSUE_TEMPLATE │ ├── bugs.md │ ├── config.yml │ ├── documentation.md │ ├── feature-request.md │ └── unexpected-problems-bugs.md ├── pull_request_template.md └── workflows │ ├── check-template.yml │ ├── levenshtein.js │ ├── needs-reply.yml │ ├── remove-needs-reply.yml │ └── workflow.yml ├── .gitignore ├── LICENSE ├── README.md ├── centernet ├── __init__.py ├── config.py ├── data │ ├── custom_build_augmentation.py │ ├── custom_dataset_dataloader.py │ ├── datasets │ │ ├── coco.py │ │ ├── nuimages.py │ │ └── objects365.py │ └── transforms │ │ ├── custom_augmentation_impl.py │ │ └── custom_transform.py └── modeling │ ├── backbone │ ├── bifpn.py │ ├── bifpn_fcos.py │ ├── dla.py │ ├── dlafpn.py │ ├── fpn_p5.py │ └── res2net.py │ ├── debug.py │ ├── dense_heads │ ├── centernet.py │ ├── centernet_head.py │ └── utils.py │ ├── layers │ ├── deform_conv.py │ ├── heatmap_focal_loss.py │ ├── iou_loss.py │ └── ml_nms.py │ ├── meta_arch │ └── centernet_detector.py │ └── roi_heads │ ├── custom_fast_rcnn.py │ ├── custom_roi_heads.py │ └── fed_loss.py ├── configs ├── Base-CenterNet-FPN.yaml ├── Base-CenterNet2.yaml ├── Base_S4_DLA.yaml ├── CenterNet-FPN_R50_1x.yaml ├── CenterNet-S4_DLA_8x.yaml ├── CenterNet2-F_R50_1x.yaml ├── CenterNet2_DLA-BiFPN-P3_24x.yaml ├── CenterNet2_DLA-BiFPN-P3_4x.yaml ├── CenterNet2_DLA-BiFPN-P5_640_16x.yaml ├── CenterNet2_DLA-BiFPN-P5_640_16x_ST.yaml ├── CenterNet2_DLA-fcosBiFPN-P5_640_16x_ST.yaml ├── CenterNet2_R2-101-DCN-BiFPN_1280_4x.yaml ├── CenterNet2_R2-101-DCN-BiFPN_4x+4x_1560_ST.yaml ├── CenterNet2_R2-101-DCN_896_4x.yaml ├── CenterNet2_R50_1x.yaml ├── CenterNet2_X101-DCN_2x.yaml ├── LVIS_CenterNet2_R50_1x.yaml ├── LVIS_CenterNet2_R50_Fed_1x.yaml ├── O365_CenterNet2_R50_1x.yaml └── nuImages_CenterNet2_DLA_640_8x.yaml ├── datasets └── README.md ├── demo.py ├── docs ├── MODEL_ZOO.md └── centernet2_teaser.jpg ├── predictor.py ├── requirements.txt ├── tools ├── README.md ├── __init__.py ├── analyze_model.py ├── benchmark.py ├── convert-torchvision-to-d2.py ├── deploy │ ├── CMakeLists.txt │ ├── README.md │ ├── export_model.py │ └── torchscript_mask_rcnn.cpp ├── lazyconfig_train_net.py ├── lightning_train_net.py ├── plain_train_net.py ├── train_net.py ├── visualize_data.py └── visualize_json_results.py └── train_net.py /.github/CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/.github/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /.github/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/.github/CONTRIBUTING.md -------------------------------------------------------------------------------- /.github/Detectron2-Logo-Horz.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/.github/Detectron2-Logo-Horz.svg -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/.github/ISSUE_TEMPLATE.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bugs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/.github/ISSUE_TEMPLATE/bugs.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/.github/ISSUE_TEMPLATE/config.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/documentation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/.github/ISSUE_TEMPLATE/documentation.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature-request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/.github/ISSUE_TEMPLATE/feature-request.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/unexpected-problems-bugs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/.github/ISSUE_TEMPLATE/unexpected-problems-bugs.md -------------------------------------------------------------------------------- /.github/pull_request_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/.github/pull_request_template.md -------------------------------------------------------------------------------- /.github/workflows/check-template.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/.github/workflows/check-template.yml -------------------------------------------------------------------------------- /.github/workflows/levenshtein.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/.github/workflows/levenshtein.js -------------------------------------------------------------------------------- /.github/workflows/needs-reply.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/.github/workflows/needs-reply.yml -------------------------------------------------------------------------------- /.github/workflows/remove-needs-reply.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/.github/workflows/remove-needs-reply.yml -------------------------------------------------------------------------------- /.github/workflows/workflow.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/.github/workflows/workflow.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/README.md -------------------------------------------------------------------------------- /centernet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/__init__.py -------------------------------------------------------------------------------- /centernet/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/config.py -------------------------------------------------------------------------------- /centernet/data/custom_build_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/data/custom_build_augmentation.py -------------------------------------------------------------------------------- /centernet/data/custom_dataset_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/data/custom_dataset_dataloader.py -------------------------------------------------------------------------------- /centernet/data/datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/data/datasets/coco.py -------------------------------------------------------------------------------- /centernet/data/datasets/nuimages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/data/datasets/nuimages.py -------------------------------------------------------------------------------- /centernet/data/datasets/objects365.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/data/datasets/objects365.py -------------------------------------------------------------------------------- /centernet/data/transforms/custom_augmentation_impl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/data/transforms/custom_augmentation_impl.py -------------------------------------------------------------------------------- /centernet/data/transforms/custom_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/data/transforms/custom_transform.py -------------------------------------------------------------------------------- /centernet/modeling/backbone/bifpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/modeling/backbone/bifpn.py -------------------------------------------------------------------------------- /centernet/modeling/backbone/bifpn_fcos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/modeling/backbone/bifpn_fcos.py -------------------------------------------------------------------------------- /centernet/modeling/backbone/dla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/modeling/backbone/dla.py -------------------------------------------------------------------------------- /centernet/modeling/backbone/dlafpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/modeling/backbone/dlafpn.py -------------------------------------------------------------------------------- /centernet/modeling/backbone/fpn_p5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/modeling/backbone/fpn_p5.py -------------------------------------------------------------------------------- /centernet/modeling/backbone/res2net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/modeling/backbone/res2net.py -------------------------------------------------------------------------------- /centernet/modeling/debug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/modeling/debug.py -------------------------------------------------------------------------------- /centernet/modeling/dense_heads/centernet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/modeling/dense_heads/centernet.py -------------------------------------------------------------------------------- /centernet/modeling/dense_heads/centernet_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/modeling/dense_heads/centernet_head.py -------------------------------------------------------------------------------- /centernet/modeling/dense_heads/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/modeling/dense_heads/utils.py -------------------------------------------------------------------------------- /centernet/modeling/layers/deform_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/modeling/layers/deform_conv.py -------------------------------------------------------------------------------- /centernet/modeling/layers/heatmap_focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/modeling/layers/heatmap_focal_loss.py -------------------------------------------------------------------------------- /centernet/modeling/layers/iou_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/modeling/layers/iou_loss.py -------------------------------------------------------------------------------- /centernet/modeling/layers/ml_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/modeling/layers/ml_nms.py -------------------------------------------------------------------------------- /centernet/modeling/meta_arch/centernet_detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/modeling/meta_arch/centernet_detector.py -------------------------------------------------------------------------------- /centernet/modeling/roi_heads/custom_fast_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/modeling/roi_heads/custom_fast_rcnn.py -------------------------------------------------------------------------------- /centernet/modeling/roi_heads/custom_roi_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/modeling/roi_heads/custom_roi_heads.py -------------------------------------------------------------------------------- /centernet/modeling/roi_heads/fed_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/centernet/modeling/roi_heads/fed_loss.py -------------------------------------------------------------------------------- /configs/Base-CenterNet-FPN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/configs/Base-CenterNet-FPN.yaml -------------------------------------------------------------------------------- /configs/Base-CenterNet2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/configs/Base-CenterNet2.yaml -------------------------------------------------------------------------------- /configs/Base_S4_DLA.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/configs/Base_S4_DLA.yaml -------------------------------------------------------------------------------- /configs/CenterNet-FPN_R50_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/configs/CenterNet-FPN_R50_1x.yaml -------------------------------------------------------------------------------- /configs/CenterNet-S4_DLA_8x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/configs/CenterNet-S4_DLA_8x.yaml -------------------------------------------------------------------------------- /configs/CenterNet2-F_R50_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/configs/CenterNet2-F_R50_1x.yaml -------------------------------------------------------------------------------- /configs/CenterNet2_DLA-BiFPN-P3_24x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/configs/CenterNet2_DLA-BiFPN-P3_24x.yaml -------------------------------------------------------------------------------- /configs/CenterNet2_DLA-BiFPN-P3_4x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/configs/CenterNet2_DLA-BiFPN-P3_4x.yaml -------------------------------------------------------------------------------- /configs/CenterNet2_DLA-BiFPN-P5_640_16x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/configs/CenterNet2_DLA-BiFPN-P5_640_16x.yaml -------------------------------------------------------------------------------- /configs/CenterNet2_DLA-BiFPN-P5_640_16x_ST.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/configs/CenterNet2_DLA-BiFPN-P5_640_16x_ST.yaml -------------------------------------------------------------------------------- /configs/CenterNet2_DLA-fcosBiFPN-P5_640_16x_ST.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/configs/CenterNet2_DLA-fcosBiFPN-P5_640_16x_ST.yaml -------------------------------------------------------------------------------- /configs/CenterNet2_R2-101-DCN-BiFPN_1280_4x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/configs/CenterNet2_R2-101-DCN-BiFPN_1280_4x.yaml -------------------------------------------------------------------------------- /configs/CenterNet2_R2-101-DCN-BiFPN_4x+4x_1560_ST.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/configs/CenterNet2_R2-101-DCN-BiFPN_4x+4x_1560_ST.yaml -------------------------------------------------------------------------------- /configs/CenterNet2_R2-101-DCN_896_4x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/configs/CenterNet2_R2-101-DCN_896_4x.yaml -------------------------------------------------------------------------------- /configs/CenterNet2_R50_1x.yaml: -------------------------------------------------------------------------------- 1 | _BASE_: "Base-CenterNet2.yaml" 2 | -------------------------------------------------------------------------------- /configs/CenterNet2_X101-DCN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/configs/CenterNet2_X101-DCN_2x.yaml -------------------------------------------------------------------------------- /configs/LVIS_CenterNet2_R50_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/configs/LVIS_CenterNet2_R50_1x.yaml -------------------------------------------------------------------------------- /configs/LVIS_CenterNet2_R50_Fed_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/configs/LVIS_CenterNet2_R50_Fed_1x.yaml -------------------------------------------------------------------------------- /configs/O365_CenterNet2_R50_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/configs/O365_CenterNet2_R50_1x.yaml -------------------------------------------------------------------------------- /configs/nuImages_CenterNet2_DLA_640_8x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/configs/nuImages_CenterNet2_DLA_640_8x.yaml -------------------------------------------------------------------------------- /datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/datasets/README.md -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/demo.py -------------------------------------------------------------------------------- /docs/MODEL_ZOO.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/docs/MODEL_ZOO.md -------------------------------------------------------------------------------- /docs/centernet2_teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/docs/centernet2_teaser.jpg -------------------------------------------------------------------------------- /predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/predictor.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | opencv-python 2 | -------------------------------------------------------------------------------- /tools/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/tools/README.md -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/analyze_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/tools/analyze_model.py -------------------------------------------------------------------------------- /tools/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/tools/benchmark.py -------------------------------------------------------------------------------- /tools/convert-torchvision-to-d2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/tools/convert-torchvision-to-d2.py -------------------------------------------------------------------------------- /tools/deploy/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/tools/deploy/CMakeLists.txt -------------------------------------------------------------------------------- /tools/deploy/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/tools/deploy/README.md -------------------------------------------------------------------------------- /tools/deploy/export_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/tools/deploy/export_model.py -------------------------------------------------------------------------------- /tools/deploy/torchscript_mask_rcnn.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/tools/deploy/torchscript_mask_rcnn.cpp -------------------------------------------------------------------------------- /tools/lazyconfig_train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/tools/lazyconfig_train_net.py -------------------------------------------------------------------------------- /tools/lightning_train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/tools/lightning_train_net.py -------------------------------------------------------------------------------- /tools/plain_train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/tools/plain_train_net.py -------------------------------------------------------------------------------- /tools/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/tools/train_net.py -------------------------------------------------------------------------------- /tools/visualize_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/tools/visualize_data.py -------------------------------------------------------------------------------- /tools/visualize_json_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/tools/visualize_json_results.py -------------------------------------------------------------------------------- /train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xingyizhou/CenterNet2/HEAD/train_net.py --------------------------------------------------------------------------------