Final-Project-Face-Enhancement

Final Project for image processing 1/2023
Member:

• Werapat Wangrungroj
• Nattapong Anansomsin

Project Overview

เนื่องจากว่าใน social media มักจะมีการลงรูปของตัวเองและหลายๆคนมักจะใช้ filter หรือการ enhance รูปใบหน้าของตัวเองก่อนลง social media จึงเป็นแรงบันดาลใจให้เราอยากทำ face enhancement

Architecture

กลุ่มของเราใช้เป็น Unet Architecture สำหรับการ segmentation ในส่วนใบหน้า และใช้ face detection จาก library dlib

Encoder	Weights	Params,M
resnext101_32x48d	instagram	826M

Process in Our Method

How to use our method

การ enhancement ใบหน้าเราจำเป็นต้องรัน code ใน file ที่สามารถ run ได้ใน google colab โดยต้อง run code cell จนถึงบรรทัดสุดท้าย และเราจึงจะสามารถเรียกใช้

enhancer(image)

ที่รับ input เป็น BGR image โดยมี output เป็น RGB image นอกจากนี้ก่อนเรียก function enhance จำเป็นต้องใช้คำสั่งนี้ครอบก่อน

model.eval()
with torch.no_grad():
 ....
 result = enhancer(image)

เพื่อเปลี่ยน model เป็น mode evaluation จะเป็นการเพิ่ม accuracy ให้ model

reference for Dataset:

Dataset: Dataset github

Dataset paper: A New Dataset and Boundary-Attention Semantic Segmentation for Face Parsing. Yinglu Liu, Hailin Shi, Hao Shen, Yue Si, Xiaobo Wang, Tao Mei. In AAAI, 2020.

Link and Assets

train/dev/test set for unet
data set that we used
data set for testing our model
data set for testing (facial mask)
data set for testing (face mask)
Model
dlib model
Medium
Data Label App
Slide Presentation