Image Unavailable (Paint)

Painted Faces

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.

Image Unavailable (Painted Faces)
Painted Faces

YOLOv5 GitHub Code

YOLOv5 Website

CUT GitHub Code

CUT Paper

CUT Website

CelebA Paper

CelebA Website

WIDER FACE Paper

WIDER FACE Website