Tooth detection and labeling

A non-transitory computer readable storage medium comprising instructions that, when executed by a processing device, cause the processing device to perform operations comprising:

• receiving an image of a face;
• processing the image using a first trained machine learning model to determine a bounding shape around teeth in the image;
• cropping the image based on the bounding shape to produce a cropped image;
• processing the cropped image using an edge detection operation to generate edge data for the cropped image;
• and
• processing the cropped image and the edge data using a second trained machine learning model to label edges in the cropped image.

A computing device comprising:

• a memory;
• and
• a processing device operatively coupled to the memory, the processing device to:
• receive an image of a face;
• process the image using a first trained machine learning model to determine a bounding shape around teeth in the image;
• crop the image based on the bounding shape to produce a cropped image;
• process the cropped image using an edge detection operation to generate edge data for the cropped image;
• and
• process the cropped image and the edge data using a second trained machine learning model to label edges in the cropped image.