/Image-Segmentation-on-CelebAMask-HQ-dataset

Performing Image Segmentation using U-Net to detect facial features like, eyes, lips, ears

Primary LanguageJupyter Notebook

Image-Segmentation-on-CelebAMask-HQ-dataset

Applied a U-Net architecture to perform image segmentation. I trained on Google Colab. Due to lack of computational resources, I only trained on 685 images.