👶🏻 MLOps를 활용한 2세 얼굴 예측 서비스

🍼 Project Summary

• StyleGAN 기반의 생성 모델인 BabyGAN을 사용하여 두 사람의 얼굴을 입력으로 받아 2세 얼굴을 합성하는 서비스입니다.
• 모델 서빙과 배포 경험을 위한 프로젝트이며, FastAPI, Airflow, AWS, Docker 를 사용하였습니다.

🍼 Repository Structure

• Front-end, Back-end, Airflow, Docker

├── README.md
├── .env
├── airflow
│   └── dag
├── back-end
│   ├── api
│   │   ├── controller
│   │   ├── model
│   │   └── service
│   ├── babygan
│   └── main
├── docker
└── front-end

🍼 Architecture Flow Map

🍼 Demo

데모링크 (12월 25일 이후로 babygan의 inference 기능이 중단됩니다.)

• Dashboard : airflow의 batchjob을 분석한 결과를 보여주는 페이지
• BabyGAN : 사용자의 입력 이미지에 따른 인퍼런스 결과물을 보여주는 페이지
  • input
  * output & share
• Gallery : 사용자가 공유한 결과물을 모아 볼 수 있는 페이지

🍼 Contributors

Thanks goes to these wonderful people (emoji key):

Jihun Heo 📆 💻 📖

Hyundong Jo 💻 📖

Sejong Jeong 💻 📖

Yujin Lee 💻 📖

Hanbin Lee 💻 📖

Seowon Kim 💻 📖

This project follows the all-contributors specification. Contributions of any kind welcome!

🍼 Reference

• BabyGAN
• FastAPI