Nevermind, I figured I need an object detection model for my use case.
Topic:
Machine Learning & AI
SubTopic:
Create ML