sharif-apu / PyTorch-Pose-Estimation

PyTorch 기반 Pose Estimation 모델 구조 및 학습 기법을 테스트하기 위한 프로젝트

PyTorch Pose Estimation

PyTorch 기반 Pose Estimation 모델 구조 및 학습 기법을 테스트하기 위한 프로젝트

Implementations

• Simple Baselines for Human Pose Estimation

TODOs

• Simple Baselines for Human Pose Estimation
• Single-Stage Multi-Person Pose Machines

Requirements

• PyTorch >= 1.8.1
• PyTorch Lightning
• Albumentations
• PyYaml
• Pycocotools

Train Detector

python train_sbp.py --cfg configs/sbp_coco.yaml

Test Detector

python test_sbp.py --cfg configs/sbp_coco.yaml

Inference Detector

python inference_sbp.py --cfg configs/sbp_coco.yaml --ckpt path/to/ckpt_file

Training Experiments

개요

• Pose Estimation 모델 훈련
• Pretrained Weights 사용하지 않고 처음부터 훈련

데이터셋

Train Dataset	Validation Dataset
COCO2017 Train	COCO2017 Validation
64115 images	2693 images

결과

• Validation Dataset 결과
• Confidence Threshold = 0.25

Method	Backbone	Input size	AP@.5
Simple Baselines	darknet19	256 x 192	80.9