A method includes determining depth values associated with a first set of pixel locations in a first image of a mouth. A first function is generated for a color channel based on intensities of the color channel at the first set of pixel locations and depth values associated with the first set of pixel locations. Image data comprising a new representation of the teeth is received, wherein the image data comprises a second set of pixel locations and new depth values associated with the second set of pixel locations. A new image is generated based on the image data and the first function, wherein a shape of the teeth is based on the image data and a color of the teeth is based on applying the first function to the second set of pixel locations for the teeth and the new depth values.