포토샵은 이제 그만, 대세는 MaskGAN이다

작성자 이원준
학과 또는 소속(회사명) 소프트웨어융합학과
조회수 26
평가(좋아요)수 10
댓글수 0
[개요]
현대 사회에 보정 어플은 상당히 발전되어있다. 하지만 기능이 뷰티에 초점이 맞춰져 있다는 한계점이 있다. 따라서 본 프로젝트에서는 얼굴 커스터마이징뿐만 아니라 헤어스타일, 귀걸이, 옷, 모자, 안경 등 다양하게 적용할 수 있는 프로젝트를 구현하고자 하였다. MaskGAN을 이용하여 사용자 친화적인 GUI 형태의 프로그램을 오픈된 코드와 논문을 바탕으로 학습 코드와 최종 Demo를 구현하였다.

[Dataset]
CelebA 데이터셋 30000장 + 팀원 4명의 사진 추가 학습

[기능]
각 기능에 맞춰 얼굴 커스터마이징이 가능하다.
-Skin
-nose
-Eyeglass
-Left Eye, Right Eye
-Left Eyebrow, Right Eyebrow
-Left ear, Right ear
-Mouth, Upper Lib, Lower Lib
-Hair
-Hat
-Earring
-Necklace
-Neck
-Cloth

* CelebA 데이터로 사전 학습된 MaskGAN 모델에 각 팀원들 사진과 mask 이미지를 추가하여 continue_train을 수행하였다.

* 제공하는 train 코드가 없어 Pix2PixHD train 코드를 참고하여 MaskGAN에 적합하게 train 코드를 짜서 진행하였다.

* 참고한 코드의 라이브러리 버전이 구 버전이여 현재 최신 버전의 라이브러리 버전에 맞게 코드를 수정하여 진행하였다.

* 자세한 수정 및 추가 사항이나 코드는 깃허브를 참고

깃허브 : https://github.com/velpegor/Face_customizing
YouTube 데모영상 : https://youtu.be/W-l0jjOcYGE

포토샵은 이제 그만, 대세는 MaskGAN이다

조회수 26
평가(좋아요)수 10
댓글수 0
게시 : 2022년 11월 10일
이원준 소프트웨어융합학과

포토샵은 이제 그만, 대세는 MaskGAN이다

조회수 26
평가(좋아요)수 10
댓글수 0
게시 : 2022-11-10

Member

이원준, 이승현, 명승준, 임달홍

Keyword

MaskGAN, GAN, 딥러닝, 2022SW페스티벌

  • 인공지능
  • 데이터분석