Retraining the human pose estimation model