/ Machine Learning

Real time multi-person pose estimation based on atrous convolution

Real time multi-person pose estimation based on atrous convolution

AtrousPose

real time multi-person pose estimation based on atrous convolution.

Network Structures

basicNet2

Spatial pyramid network for prediction

spatialpyramid

Properties of atrouspose network

Arch Input Size Output Size FLOPS Num.param. FPS(C++)
AtrousPose-512 384×384 48×48 50G 26M 42
AtrousPose-128 384×384 48×48 20G 13M 67

Test Results

demo2

Evaluation

Demo

  • (Python demo)run > demo.py results will be store in /images
  • (C++ version)

Training

  • download mpii dataset
  • download mask for mpii
  • download json for mpii
  • train
    • run > train.py

loss

GitHub