├── .gitignore ├── .gitmodules ├── LICENSE ├── README.md ├── README_ORIGINAL.md ├── configs ├── _base_ │ ├── datasets │ │ ├── coco_instance.py │ │ └── coco_keypoint.py │ └── default_runtime.py ├── inspose │ ├── README.md │ └── inspose_r50_8x4_3x_coco.py ├── petr │ ├── README.md │ ├── petr_r101_8x1_100e_coco_custom.py │ ├── petr_r101_8x1_100e_coco_t900_group5_ffn2x_dp0_custom.py │ ├── petr_r50_8x1_100e_coco_custom.py │ ├── petr_r50_8x1_100e_coco_t900_group5_ffn2x_dp0_custom.py │ ├── petr_swin-l-p4-w7-22kto1k_8x1_100e_coco_custom_droppath0.5.py │ └── petr_swin-l-p4-w7-22kto1k_8x1_100e_coco_t900_group5_ffn2x_dp0_custom_droppath0.5.py └── soit │ ├── README.md │ ├── soit_r101_16x2_50e_coco.py │ └── soit_r50_16x2_50e_coco.py ├── docs └── get_started.md ├── opera ├── __init__.py ├── apis │ ├── __init__.py │ ├── inference.py │ ├── test.py │ └── train.py ├── core │ ├── __init__.py │ ├── bbox │ │ ├── __init__.py │ │ ├── assigners │ │ │ ├── __init__.py │ │ │ └── hungarian_assigner.py │ │ ├── builder.py │ │ └── match_costs │ │ │ ├── __init__.py │ │ │ ├── builder.py │ │ │ └── match_cost.py │ ├── evaluation │ │ ├── __init__.py │ │ └── eval_hooks.py │ ├── keypoint │ │ ├── __init__.py │ │ └── transforms.py │ ├── post_processing │ │ └── __init__.py │ └── runner │ │ └── __init__.py ├── datasets │ ├── __init__.py │ ├── builder.py │ ├── coco_pose.py │ └── pipelines │ │ ├── __init__.py │ │ ├── formating.py │ │ ├── loading.py │ │ └── transforms.py ├── models │ ├── __init__.py │ ├── backbones │ │ └── __init__.py │ ├── builder.py │ ├── dense_heads │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ └── petr_head.py │ │ ├── inspose_head.py │ │ ├── ori_petr_head.py │ │ ├── petr_head.py │ │ └── soit_head.py │ ├── detectors │ │ ├── __init__.py │ │ ├── inspose.py │ │ ├── petr.py │ │ └── soit.py │ ├── losses │ │ ├── __init__.py │ │ └── oks_loss.py │ ├── necks │ │ └── __init__.py │ ├── roi_heads │ │ └── __init__.py │ └── utils │ │ ├── __init__.py │ │ ├── builder.py │ │ ├── positional_encoding.py │ │ └── transformer.py └── version.py ├── requirements.txt ├── requirements ├── build.txt ├── docs.txt ├── mminstall.txt ├── optional.txt ├── readthedocs.txt ├── runtime.txt └── tests.txt ├── setup.cfg ├── setup.py └── tools ├── dataset_converters └── keypoint2pesudo_box.py ├── dist_test.sh ├── dist_train.sh ├── eval_metric.py ├── test.py └── train.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ 2 | *.pyc -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/.gitmodules -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/README.md -------------------------------------------------------------------------------- /README_ORIGINAL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/README_ORIGINAL.md -------------------------------------------------------------------------------- /configs/_base_/datasets/coco_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/configs/_base_/datasets/coco_instance.py -------------------------------------------------------------------------------- /configs/_base_/datasets/coco_keypoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/configs/_base_/datasets/coco_keypoint.py -------------------------------------------------------------------------------- /configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /configs/inspose/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/configs/inspose/README.md -------------------------------------------------------------------------------- /configs/inspose/inspose_r50_8x4_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/configs/inspose/inspose_r50_8x4_3x_coco.py -------------------------------------------------------------------------------- /configs/petr/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/configs/petr/README.md -------------------------------------------------------------------------------- /configs/petr/petr_r101_8x1_100e_coco_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/configs/petr/petr_r101_8x1_100e_coco_custom.py -------------------------------------------------------------------------------- /configs/petr/petr_r101_8x1_100e_coco_t900_group5_ffn2x_dp0_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/configs/petr/petr_r101_8x1_100e_coco_t900_group5_ffn2x_dp0_custom.py -------------------------------------------------------------------------------- /configs/petr/petr_r50_8x1_100e_coco_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/configs/petr/petr_r50_8x1_100e_coco_custom.py -------------------------------------------------------------------------------- /configs/petr/petr_r50_8x1_100e_coco_t900_group5_ffn2x_dp0_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/configs/petr/petr_r50_8x1_100e_coco_t900_group5_ffn2x_dp0_custom.py -------------------------------------------------------------------------------- /configs/petr/petr_swin-l-p4-w7-22kto1k_8x1_100e_coco_custom_droppath0.5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/configs/petr/petr_swin-l-p4-w7-22kto1k_8x1_100e_coco_custom_droppath0.5.py -------------------------------------------------------------------------------- /configs/petr/petr_swin-l-p4-w7-22kto1k_8x1_100e_coco_t900_group5_ffn2x_dp0_custom_droppath0.5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/configs/petr/petr_swin-l-p4-w7-22kto1k_8x1_100e_coco_t900_group5_ffn2x_dp0_custom_droppath0.5.py -------------------------------------------------------------------------------- /configs/soit/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/configs/soit/README.md -------------------------------------------------------------------------------- /configs/soit/soit_r101_16x2_50e_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/configs/soit/soit_r101_16x2_50e_coco.py -------------------------------------------------------------------------------- /configs/soit/soit_r50_16x2_50e_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/configs/soit/soit_r50_16x2_50e_coco.py -------------------------------------------------------------------------------- /docs/get_started.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/docs/get_started.md -------------------------------------------------------------------------------- /opera/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/__init__.py -------------------------------------------------------------------------------- /opera/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/apis/__init__.py -------------------------------------------------------------------------------- /opera/apis/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/apis/inference.py -------------------------------------------------------------------------------- /opera/apis/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/apis/test.py -------------------------------------------------------------------------------- /opera/apis/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/apis/train.py -------------------------------------------------------------------------------- /opera/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/core/__init__.py -------------------------------------------------------------------------------- /opera/core/bbox/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/core/bbox/__init__.py -------------------------------------------------------------------------------- /opera/core/bbox/assigners/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/core/bbox/assigners/__init__.py -------------------------------------------------------------------------------- /opera/core/bbox/assigners/hungarian_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/core/bbox/assigners/hungarian_assigner.py -------------------------------------------------------------------------------- /opera/core/bbox/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/core/bbox/builder.py -------------------------------------------------------------------------------- /opera/core/bbox/match_costs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/core/bbox/match_costs/__init__.py -------------------------------------------------------------------------------- /opera/core/bbox/match_costs/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/core/bbox/match_costs/builder.py -------------------------------------------------------------------------------- /opera/core/bbox/match_costs/match_cost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/core/bbox/match_costs/match_cost.py -------------------------------------------------------------------------------- /opera/core/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/core/evaluation/__init__.py -------------------------------------------------------------------------------- /opera/core/evaluation/eval_hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/core/evaluation/eval_hooks.py -------------------------------------------------------------------------------- /opera/core/keypoint/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/core/keypoint/__init__.py -------------------------------------------------------------------------------- /opera/core/keypoint/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/core/keypoint/transforms.py -------------------------------------------------------------------------------- /opera/core/post_processing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/core/post_processing/__init__.py -------------------------------------------------------------------------------- /opera/core/runner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/core/runner/__init__.py -------------------------------------------------------------------------------- /opera/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/datasets/__init__.py -------------------------------------------------------------------------------- /opera/datasets/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/datasets/builder.py -------------------------------------------------------------------------------- /opera/datasets/coco_pose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/datasets/coco_pose.py -------------------------------------------------------------------------------- /opera/datasets/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/datasets/pipelines/__init__.py -------------------------------------------------------------------------------- /opera/datasets/pipelines/formating.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/datasets/pipelines/formating.py -------------------------------------------------------------------------------- /opera/datasets/pipelines/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/datasets/pipelines/loading.py -------------------------------------------------------------------------------- /opera/datasets/pipelines/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/datasets/pipelines/transforms.py -------------------------------------------------------------------------------- /opera/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/models/__init__.py -------------------------------------------------------------------------------- /opera/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/models/backbones/__init__.py -------------------------------------------------------------------------------- /opera/models/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/models/builder.py -------------------------------------------------------------------------------- /opera/models/dense_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/models/dense_heads/__init__.py -------------------------------------------------------------------------------- /opera/models/dense_heads/__pycache__/petr_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/models/dense_heads/__pycache__/petr_head.py -------------------------------------------------------------------------------- /opera/models/dense_heads/inspose_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/models/dense_heads/inspose_head.py -------------------------------------------------------------------------------- /opera/models/dense_heads/ori_petr_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/models/dense_heads/ori_petr_head.py -------------------------------------------------------------------------------- /opera/models/dense_heads/petr_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/models/dense_heads/petr_head.py -------------------------------------------------------------------------------- /opera/models/dense_heads/soit_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/models/dense_heads/soit_head.py -------------------------------------------------------------------------------- /opera/models/detectors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/models/detectors/__init__.py -------------------------------------------------------------------------------- /opera/models/detectors/inspose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/models/detectors/inspose.py -------------------------------------------------------------------------------- /opera/models/detectors/petr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/models/detectors/petr.py -------------------------------------------------------------------------------- /opera/models/detectors/soit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/models/detectors/soit.py -------------------------------------------------------------------------------- /opera/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/models/losses/__init__.py -------------------------------------------------------------------------------- /opera/models/losses/oks_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/models/losses/oks_loss.py -------------------------------------------------------------------------------- /opera/models/necks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/models/necks/__init__.py -------------------------------------------------------------------------------- /opera/models/roi_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/models/roi_heads/__init__.py -------------------------------------------------------------------------------- /opera/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/models/utils/__init__.py -------------------------------------------------------------------------------- /opera/models/utils/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/models/utils/builder.py -------------------------------------------------------------------------------- /opera/models/utils/positional_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/models/utils/positional_encoding.py -------------------------------------------------------------------------------- /opera/models/utils/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/models/utils/transformer.py -------------------------------------------------------------------------------- /opera/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/opera/version.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/requirements.txt -------------------------------------------------------------------------------- /requirements/build.txt: -------------------------------------------------------------------------------- 1 | # These must be installed before building opera 2 | cython 3 | numpy 4 | -------------------------------------------------------------------------------- /requirements/docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/requirements/docs.txt -------------------------------------------------------------------------------- /requirements/mminstall.txt: -------------------------------------------------------------------------------- 1 | mmcv-full>=1.3.17 2 | -------------------------------------------------------------------------------- /requirements/optional.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/requirements/optional.txt -------------------------------------------------------------------------------- /requirements/readthedocs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/requirements/readthedocs.txt -------------------------------------------------------------------------------- /requirements/runtime.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/requirements/runtime.txt -------------------------------------------------------------------------------- /requirements/tests.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/requirements/tests.txt -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/setup.py -------------------------------------------------------------------------------- /tools/dataset_converters/keypoint2pesudo_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/tools/dataset_converters/keypoint2pesudo_box.py -------------------------------------------------------------------------------- /tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/tools/dist_test.sh -------------------------------------------------------------------------------- /tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/tools/dist_train.sh -------------------------------------------------------------------------------- /tools/eval_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/tools/eval_metric.py -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HDETR/H-PETR-Pose/HEAD/tools/train.py --------------------------------------------------------------------------------