├── .gitignore ├── .gitmodules ├── LICENSE ├── README.md ├── data ├── acc_curve.png ├── lsp │ └── images │ │ ├── lsp_dataset │ │ └── lspet_dataset ├── mpii │ ├── mean.pth.tar │ └── mpii_annotations.json └── mscoco │ └── README.md ├── evaluation ├── data │ ├── detections.mat │ └── detections_our_format.mat ├── eval_PCKh.m ├── eval_PCKh.py ├── showskeletons_joints.m └── utils.py ├── example ├── lsp.py ├── mpii.py └── mscoco.py ├── miscs ├── cocoScale.m ├── gen_coco.m ├── gen_lsp.m └── gen_mpii.m ├── pose ├── __init__.py ├── datasets │ ├── __init__.py │ ├── lsp.py │ ├── mpii.py │ └── mscoco.py ├── models │ ├── __init__.py │ ├── hourglass.py │ └── preresnet.py └── utils │ ├── __init__.py │ ├── evaluation.py │ ├── imutils.py │ ├── logger.py │ ├── misc.py │ ├── osutils.py │ └── transforms.py ├── requirements.txt └── tools ├── mpii_demo.py └── mpii_export_to_onxx.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/.gitmodules -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/README.md -------------------------------------------------------------------------------- /data/acc_curve.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/data/acc_curve.png -------------------------------------------------------------------------------- /data/lsp/images/lsp_dataset: -------------------------------------------------------------------------------- 1 | /home/wyang/Data/dataset/LSP -------------------------------------------------------------------------------- /data/lsp/images/lspet_dataset: -------------------------------------------------------------------------------- 1 | /home/wyang/Data/dataset/LSP_ext/ -------------------------------------------------------------------------------- /data/mpii/mean.pth.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/data/mpii/mean.pth.tar -------------------------------------------------------------------------------- /data/mpii/mpii_annotations.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/data/mpii/mpii_annotations.json -------------------------------------------------------------------------------- /data/mscoco/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/data/mscoco/README.md -------------------------------------------------------------------------------- /evaluation/data/detections.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/evaluation/data/detections.mat -------------------------------------------------------------------------------- /evaluation/data/detections_our_format.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/evaluation/data/detections_our_format.mat -------------------------------------------------------------------------------- /evaluation/eval_PCKh.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/evaluation/eval_PCKh.m -------------------------------------------------------------------------------- /evaluation/eval_PCKh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/evaluation/eval_PCKh.py -------------------------------------------------------------------------------- /evaluation/showskeletons_joints.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/evaluation/showskeletons_joints.m -------------------------------------------------------------------------------- /evaluation/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/evaluation/utils.py -------------------------------------------------------------------------------- /example/lsp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/example/lsp.py -------------------------------------------------------------------------------- /example/mpii.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/example/mpii.py -------------------------------------------------------------------------------- /example/mscoco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/example/mscoco.py -------------------------------------------------------------------------------- /miscs/cocoScale.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/miscs/cocoScale.m -------------------------------------------------------------------------------- /miscs/gen_coco.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/miscs/gen_coco.m -------------------------------------------------------------------------------- /miscs/gen_lsp.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/miscs/gen_lsp.m -------------------------------------------------------------------------------- /miscs/gen_mpii.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/miscs/gen_mpii.m -------------------------------------------------------------------------------- /pose/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/pose/__init__.py -------------------------------------------------------------------------------- /pose/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/pose/datasets/__init__.py -------------------------------------------------------------------------------- /pose/datasets/lsp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/pose/datasets/lsp.py -------------------------------------------------------------------------------- /pose/datasets/mpii.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/pose/datasets/mpii.py -------------------------------------------------------------------------------- /pose/datasets/mscoco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/pose/datasets/mscoco.py -------------------------------------------------------------------------------- /pose/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/pose/models/__init__.py -------------------------------------------------------------------------------- /pose/models/hourglass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/pose/models/hourglass.py -------------------------------------------------------------------------------- /pose/models/preresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/pose/models/preresnet.py -------------------------------------------------------------------------------- /pose/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/pose/utils/__init__.py -------------------------------------------------------------------------------- /pose/utils/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/pose/utils/evaluation.py -------------------------------------------------------------------------------- /pose/utils/imutils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/pose/utils/imutils.py -------------------------------------------------------------------------------- /pose/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/pose/utils/logger.py -------------------------------------------------------------------------------- /pose/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/pose/utils/misc.py -------------------------------------------------------------------------------- /pose/utils/osutils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/pose/utils/osutils.py -------------------------------------------------------------------------------- /pose/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/pose/utils/transforms.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/requirements.txt -------------------------------------------------------------------------------- /tools/mpii_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/tools/mpii_demo.py -------------------------------------------------------------------------------- /tools/mpii_export_to_onxx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuanyuanli85/pytorch-pose/HEAD/tools/mpii_export_to_onxx.py --------------------------------------------------------------------------------