diff --git a/docs/guidance/preparation.md b/docs/guidance/preparation.md index 63fd8a6..dd17b96 100644 --- a/docs/guidance/preparation.md +++ b/docs/guidance/preparation.md @@ -11,6 +11,7 @@ conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10 pip install insightface==0.2.1 onnxruntime moviepy ``` - We use the face detection and alignment methods from **[insightface](https://github.com/deepinsight/insightface)** for image preprocessing. Please download the relative files and unzip them to ./insightface_func/models from [this link](https://onedrive.live.com/?authkey=%21ADJ0aAOSsc90neY&cid=4A83B6B633B029CC&id=4A83B6B633B029CC%215837&parId=4A83B6B633B029CC%215834&action=locate). +- We use the face parsing from **[face-parsing.PyTorch](https://github.com/zllrunning/face-parsing.PyTorch)** for image postprocessing. Please download the relative file and place it in ./parsing_model/checkpoint from [this link](https://drive.google.com/file/d/154JgKpzCPW82qINcVieuPH3fZ2e0P812/view). - The pytorch and cuda versions above are most recommanded. They may vary. - Using insightface with different versions is not recommanded. Please use this specific version. - These settings are tested valid on both Windows and Ununtu. @@ -25,4 +26,4 @@ There are two archive files in the drive: **checkpoints.zip** and **arcface_chec [[Baidu Drive]](https://pan.baidu.com/s/1wFV11RVZMHqd-ky4YpLdcA) Password: ```jd2v``` ### Note -We expect users to have GPU with at least 8G memory. For those who do not, we provide [[Colab Notebook implementation]](https://colab.research.google.com/github/neuralchen/SimSwap/blob/main/SimSwap%20colab.ipynb). +We expect users to have GPU with at least 3G memory. For those who do not, we provide [[Colab Notebook implementation]](https://colab.research.google.com/github/neuralchen/SimSwap/blob/main/SimSwap%20colab.ipynb).