I am following the official documentation to train on a custom dataset for Human Pose Estimation from ultralytics in Python. I have a small dataset of 1 image labeled by me, and the predictions work as expected, so the vertices recognition and class recognition work fine.