Painted Faces
This code uses two AIs to extract the faces from images and then convert them into painted portraits. I have reused my YOLOv5 from the emotion detection project to extract the faces and this is then fed through the generator from a CUT model. The CUT model was trained on a custom dataset based on the CelebA dataset for the conversion task. Together, the code runs on single images or folders for batch conversion.