├── .gitmodules ├── README.md ├── configs ├── _base_ │ └── datasets │ │ └── coco.py └── body │ └── 2d_kpt_sview_rgb_img │ ├── distilpose │ └── coco │ │ ├── DistilPose_L_coco_256x192.py │ │ ├── DistilPose_S_coco_256x192.py │ │ ├── hr48_3stage_coco_256x192.py │ │ └── stemnet_coco_256x192.py │ └── tokenpose │ └── coco │ ├── teacher_hr48_coco_256x192.py │ └── teacher_stemnet_coco_256x192.py ├── demo ├── show_img_res.sh └── top_down_img_demo.py ├── distilpose ├── apis │ ├── __init__.py │ └── inference.py └── models │ ├── __init__.py │ ├── backbones │ ├── __init__.py │ ├── hrnet_3stage.py │ └── stemnet.py │ ├── detectors │ ├── __init__.py │ └── top_down_dist.py │ ├── heads │ ├── __init__.py │ ├── distilpose_head.py │ ├── tokenpose_head.py │ └── utils │ │ ├── __init__.py │ │ └── tokenbase.py │ └── losses │ ├── __init__.py │ └── dist_loss.py ├── images └── framework.png ├── requirements.txt └── tools ├── benchmark_inference.py ├── dist_test.sh ├── dist_train.sh ├── get_flops.py ├── test.py ├── test_flops.sh ├── test_fps.sh └── train.py /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/.gitmodules -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/README.md -------------------------------------------------------------------------------- /configs/_base_/datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/configs/_base_/datasets/coco.py -------------------------------------------------------------------------------- /configs/body/2d_kpt_sview_rgb_img/distilpose/coco/DistilPose_L_coco_256x192.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/configs/body/2d_kpt_sview_rgb_img/distilpose/coco/DistilPose_L_coco_256x192.py -------------------------------------------------------------------------------- /configs/body/2d_kpt_sview_rgb_img/distilpose/coco/DistilPose_S_coco_256x192.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/configs/body/2d_kpt_sview_rgb_img/distilpose/coco/DistilPose_S_coco_256x192.py -------------------------------------------------------------------------------- /configs/body/2d_kpt_sview_rgb_img/distilpose/coco/hr48_3stage_coco_256x192.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/configs/body/2d_kpt_sview_rgb_img/distilpose/coco/hr48_3stage_coco_256x192.py -------------------------------------------------------------------------------- /configs/body/2d_kpt_sview_rgb_img/distilpose/coco/stemnet_coco_256x192.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/configs/body/2d_kpt_sview_rgb_img/distilpose/coco/stemnet_coco_256x192.py -------------------------------------------------------------------------------- /configs/body/2d_kpt_sview_rgb_img/tokenpose/coco/teacher_hr48_coco_256x192.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/configs/body/2d_kpt_sview_rgb_img/tokenpose/coco/teacher_hr48_coco_256x192.py -------------------------------------------------------------------------------- /configs/body/2d_kpt_sview_rgb_img/tokenpose/coco/teacher_stemnet_coco_256x192.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/configs/body/2d_kpt_sview_rgb_img/tokenpose/coco/teacher_stemnet_coco_256x192.py -------------------------------------------------------------------------------- /demo/show_img_res.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/demo/show_img_res.sh -------------------------------------------------------------------------------- /demo/top_down_img_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/demo/top_down_img_demo.py -------------------------------------------------------------------------------- /distilpose/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/distilpose/apis/__init__.py -------------------------------------------------------------------------------- /distilpose/apis/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/distilpose/apis/inference.py -------------------------------------------------------------------------------- /distilpose/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/distilpose/models/__init__.py -------------------------------------------------------------------------------- /distilpose/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/distilpose/models/backbones/__init__.py -------------------------------------------------------------------------------- /distilpose/models/backbones/hrnet_3stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/distilpose/models/backbones/hrnet_3stage.py -------------------------------------------------------------------------------- /distilpose/models/backbones/stemnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/distilpose/models/backbones/stemnet.py -------------------------------------------------------------------------------- /distilpose/models/detectors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/distilpose/models/detectors/__init__.py -------------------------------------------------------------------------------- /distilpose/models/detectors/top_down_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/distilpose/models/detectors/top_down_dist.py -------------------------------------------------------------------------------- /distilpose/models/heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/distilpose/models/heads/__init__.py -------------------------------------------------------------------------------- /distilpose/models/heads/distilpose_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/distilpose/models/heads/distilpose_head.py -------------------------------------------------------------------------------- /distilpose/models/heads/tokenpose_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/distilpose/models/heads/tokenpose_head.py -------------------------------------------------------------------------------- /distilpose/models/heads/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /distilpose/models/heads/utils/tokenbase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/distilpose/models/heads/utils/tokenbase.py -------------------------------------------------------------------------------- /distilpose/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/distilpose/models/losses/__init__.py -------------------------------------------------------------------------------- /distilpose/models/losses/dist_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/distilpose/models/losses/dist_loss.py -------------------------------------------------------------------------------- /images/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/images/framework.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | easydict 2 | einops 3 | numpy==1.22.0 4 | xtcocotools==1.12 5 | timm -------------------------------------------------------------------------------- /tools/benchmark_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/tools/benchmark_inference.py -------------------------------------------------------------------------------- /tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/tools/dist_test.sh -------------------------------------------------------------------------------- /tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/tools/dist_train.sh -------------------------------------------------------------------------------- /tools/get_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/tools/get_flops.py -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/test_flops.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/tools/test_flops.sh -------------------------------------------------------------------------------- /tools/test_fps.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/tools/test_fps.sh -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yshMars/DistilPose/HEAD/tools/train.py --------------------------------------------------------------------------------