Keras implement of Stacked Hourglass. The model is for keypoint detection, and can be migrated to other tasks.
If you're using tf2.0 or higher, replace keras
with tf.keras
at import.
Way to build a model is exampled in StackedHourglass.py.
Full train-eval-test pipeline is exampled in test.py.
After inference OKSEvaluator.py can be used to evaluate average OKS of the result.
feiyangsuo / stacked-hourglass-for-keypoint-detection Goto Github PK
View Code? Open in Web Editor NEWKeras implement of Stacked Hourglass. The model is for keypoint detection, and can be migrated to other tasks.