sijinkim/face-swap

Implementation of basic Face-Swap/Image preprocessing.

face-swap

Face Swap base architecture 기반 모델 구현 및 코드 리팩토링 레포지토리입니다.

face-swap's Issues

apply pytorch-lightning

pytorch-lightning 적용하여 script 코드 개선

• train part
• inference part

create face data class

Face Class

• Face Detection -> Extraction 결과 정보를 담기 위함

Create main script - inference part

One script format - inference part 작성

• inference 관련 module 구현 포함

Add a Face Extractor

1. Extraction 의 세부 모듈 구현

• Dataset 설계 구현 (base FrameDataset, Transform)

• Frame 별 Face Detection 모듈

• Face Detection model

• Face Detection model을 사용하여 Frame에서 Face detect

• Detection 결과 저장 / Mask, Align 정보 핸들링 필요

Add a video to frame extractor

1. Extraction 의 세부 모듈 구현

• Video to Frame extractor

• 입력: 비디오

• 출력: Frames - 입력받은 비디오에서 추출한 이미지들을 서브 폴더를 만들어 하위에 저장한다.

• 출력된 Frames는 그들이 어느 비디오에서 추출된 것인지 추적할 수 있도록 한다.

Create main script - train part

train part script 작성

• train 관련 module 추가 및 변경 포함

create one script version

script 기반 코드 구현 우선 진행

해당 작업 후 레이어 아키텍쳐 적용합니다.

Set base architecture

Face swap workflow 중심으로 도메인 구성 잡기

1. Extraction

• Face Detection
• Face Alignment
• Face Segment

2. Training
3. Conversion