Skip to main content
3 of 3
added 25 characters in body; edited tags; edited title
nbro
  • 41.4k
  • 12
  • 114
  • 205

Which model should I use to find (only) the object location (in terms of coordinates) in an image?

I am generating images that consist of points, where the object's location is where the most overlap of points occurs.

Generated image example

In this example, the object location is $(25, 51)$.

I am trying to train a model to just finds the location, so I don't care about the classification of the object. Additionally, the shape of the overlapping points where the object is located never changes and will always be that shape.

What is a good model for this objective?

Many of the potential models I've been looking at (CNN, YOLO, and R-CNN) are more concerned with classification than location. Should I search the image for the overlapping dots, create a bound box around them, then retrieve the boxes' coordinates?

scooter
  • 41
  • 1
  • 3