Web page project based on StyleClip

Web

The Web page project is available at ~~https://wqyai.site/~~. (Due to expiration of the web server, the website is now not used, please move to https://huggingface.co/spaces/QiyuWu/StyleClip_Demo for a try) Welcome to visit my website to enjoy the fun of editing pictures. :laughing:

Environment

To run the code, you can run the following commands to download the required packages.

conda install --yes -c pytorch pytorch=1.7.1 torchvision
pip install ftfy regex tqdm gdown flask onnx onnxruntime
pip install dlib==19.6.1

Pre-trained model

The repository uploaded the code used by the front and back end of the web page, but the pre-trained model at runtime was not uploaded. When you run the project code, you need to download the pre-trained model at the link below.

Pre-trained models for specific editing effects:

| File name | Download link | | ----------------- | ------------------------------------------------------------ | | angry.pt | https://drive.google.com/uc?id=1g82HEH0jFDrcbCtn3M22gesWKfzWV_ma | | surprised.pt | https://drive.google.com/uc?id=1F-mPrhO-UeWrV1QYMZck63R43aLtPChI | | bowlcut.pt | https://drive.google.com/uc?id=1xwdxI2YCewSt05dEHgkpmmzoauPjEnnZ | | curly_hair.pt | https://drive.google.com/uc?id=1xZ7fFB12Ci6rUbUfaHPpo44xUFzpWQ6M | | purple_hair.pt | https://drive.google.com/uc?id=14H0CGXWxePrrKIYmZnDD2Ccs65EEww75 | | beyonce.pt | https://drive.google.com/uc?id=1KJTc-h02LXs4zqCyo7pzCp0iWeO6T9fz | | depp.pt | https://drive.google.com/uc?id=1FPiJkvFPG_y-bFanxLLP91wUKuy-l3IV | | hilary_clinton.pt | https://drive.google.com/uc?id=1X7U2zj2lt0KFifIsTfOOzVZXqYyCWVll | | taylor_swift.pt | https://drive.google.com/uc?id=10jHuHsKKJxuf3N0vgQbX_SMEQgFHDrZa | | trump.pt | https://drive.google.com/uc?id=14v8D0uzy4tOyfBU3ca9T0AzTt3v-dNyh | | zuckerberg.pt | https://drive.google.com/uc?id=1NjDcMUL8G-pO3i_9N6EPpQNXeMc3Ar1r | | afro.pt | https://drive.google.com/uc?id=1i5vAqo4z0I-Yon3FNft_YZOq7ClWayQJ |

Note: The above model should be downloaded and placed in the Project/notebook/mapper/pretrained.

Pre-trained models for encoder and generator:

| File name | Download link | | -------------------------- | ------------------------------------------------------------ | | e4e_ffhq_encode.pt | https://drive.google.com/file/d/1cUv_reLE6k3604or78EranS7XzuVMWeO/view?usp=sharing | | stylegan2-ffhq-config-f.pt | https://drive.google.com/file/d/1EM87UquaoQmk17Q8d5kYIAHqu0dkYqdT/view?usp=sharing |

Note: The above model should be downloaded and placed in the Project/pretrained_models.

Run the code

python start.py

The code of the project is based on the following code on github.

• orpatashnik/StyleCLIP: Official Implementation for "StyleCLIP: Text-Driven Manipulation of StyleGAN Imagery" (ICCV 2021 Oral) (github.com)

• pbaylies/stylegan-encoder: StyleGAN Encoder - converts real images to latent space (github.com)

• omertov/encoder4editing: Official implementation of "Designing an Encoder for StyleGAN Image Manipulation" (SIGGRAPH 2021) https://arxiv.org/abs/2102.02766 (github.com)

Examples

Here are some good examples of using the Web page.