Automated bounding box generation for objects in an image

摘要

One or more embodiments described herein include a computer-implemented method of determining a bounding box for an object in an image. The method includes determining a label for an object in a first image using a first algorithm, and generating a set of images based on the first image, by cropping the first image from a selected direction. The method further includes determining labels for each image in the set using the first algorithm, and removing images from the set such that the remaining images have a label matching the initial label. The method further includes determining a key image for the set, which is the smallest image from the set that has a confidence score exceeding a threshold. Further, the method includes determining a bounding box for the object in the first image based on a perimeter of a portion of the first image that overlaps the key image.