FastAPI_test

Deepfake 구현 중

---

1) Directory 구조

   fastapi_test
   |
   ├── 📁 ai
   │    ├── 📁 SOAT ( Toonification )
   │    │    ├── 📁 imgs
   |    |    |    └──  ⋮  
   |    |    ├── 📁 op
   |    |    |    └──  ⋮
   |    |    ├── 💾 projector.py ( 생성모델을 통한 사용자 이미지 인버젼 )
   |    |    ├── 💾 toonify.py ( 만화캐릭터와 합성 )
   |    |    └──  ⋮
   |    |
   │    └── 📁 Thin-Plate-Spline-Motion-Model ( Deep-fake )
   |         ├── 📁 checkpoints
   │         ├── 💾 realtime.py
   │         └──  ⋮ 
   |
   ├── 📁 app
   |    ├── 💾 __main__.py
   |    ├── 💾 main.py
   |    ├── 💾 frontend.py
   │    └──  ⋮
   |
   ├── 💾 README.md
   └── 💾 requirements.txt

2) Start Toonification

1. Python requirements

   Python : 3.10.9

   
   

2. Installation

   1. 가상 환경 설정

   2. CUDA 설정

      • https://github.com/alsqja/O-Talk/wiki/CUDA
      • 파이토치 설치 : pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
   
   

   3. 프로젝트 의존성 설치.

      • > pip install -r requirements.txt
   
   

   4. 가중치 다운로드

      모든 가중치는 ai/SOAT에 다운받기.

      • face.pt
      • disney.pt
      • inversion_stats.npz
   
   
   5. Frontend(Streamlit)와 Server(FastAPI)를 같이 실행.
      > make -j 2 run_app

참고

• https://mumin-blog.tistory.com/337