I have an image dataset with 10 different object classes and their bounding boxes. I want to train a model to predict the most probable position of the object.