Custom Pose Estimation for Mobile with Fritz AI Studio
Generate images and train your own mobile-ready pose estimation model with Fritz AI
In this article, we saw that we can use a pre-trained model from Fritz AI to perform pose estimation. Now, we will look at the steps we have to take in order to train a custom pose estimation model using Fritz Studio.
Getting Started
In this piece, we’ll train a model that is able to identify the right eye, left eye, and nose on a human face. A model that recognizes other body parts such as the elbow, ankle, etc., can also be trained, however, that requires more training images as well as more training time. For simplicity’s sake, we will use just the three key points mentioned earlier. The images used for this project can be found here and here. Before we get too far into this, you’ll first need to create an account at Fritz AI.
Once you are logged in, click Create a Project on the left panel. Then make sure you select Custom Trained in order to get the option to train your own model.