Upload 6 files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+Video.gif filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,13 +1,70 @@
----
-title: StyleClip Demo
-emoji: ⚡
-colorFrom: red
-colorTo: indigo
-sdk: gradio
-sdk_version: 4.40.0
-app_file: app.py
-pinned: false
-license: apache-2.0
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Web page project based on StyleClip
+
+### Web
+
+The Web page project is available at https://wqyai.site/. 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
+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)](https://github.com/orpatashnik/StyleCLIP)
+
+- [pbaylies/stylegan-encoder: StyleGAN Encoder - converts real images to latent space (github.com)](https://github.com/pbaylies/stylegan-encoder)
+
+- [omertov/encoder4editing: Official implementation of "Designing an Encoder for StyleGAN Image Manipulation" (SIGGRAPH 2021) https://arxiv.org/abs/2102.02766 (github.com)](https://github.com/omertov/encoder4editing)
+
+  
+
+### Examples
+
+Here are some good examples of using the Web page.
+
+![](./static/img/Snipaste_2022-08-21_21-32-17.png)
+
+![](./static/img/Snipaste_2022-08-21_21-34-26.png)

e4e.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a664ff927aa5566733a187fd29e3c842ed46bed051afb4f33dc15242a5b49b25
+size 1068794864

main.py

@@ -0,0 +1,11 @@
+from Project.aligned_image.aligned_images import align
+from Project.scripts.inference import inference
+from Project.notebook.out import out
+def final(type):
+    # @param ['afro', 'angry', 'Beyonce', 'bobcut', 'bowlcut', 'curly hair', 'Hilary Clinton', 'Jhonny Depp', 'mohawk', 'purple hair', 'surprised', 'Taylor Swift', 'trump', 'Mark Zuckerberg']
+    align()
+    inference()
+    out(type)
+
+if __name__=="__main__":
+    final('afro')

requirements.txt

@@ -0,0 +1,12 @@
+dlib==19.24.4
+Flask==2.2.5
+gradio==4.40.0
+matplotlib==3.8.4
+numpy==2.0.1
+onnx==1.16.0
+onnxruntime==1.18.1
+Pillow==10.4.0
+scipy==1.14.0
+torch==2.2.2
+torchvision==0.17.2
+tqdm==4.66.4

start.py

@@ -0,0 +1,67 @@
+import gradio as gr
+import cv2
+import numpy as np
+from Project.aligned_image.aligned_images import align
+from Project.scripts.inference import inference
+from Project.notebook.out import out
+
+def final(image, style):
+    myimg = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+    cv2.imwrite('static/img_in/in.jpg', myimg)
+    align()
+    inference()
+    out(style)
+    result = cv2.imread('static/img_out/inference_results/00000.jpg')
+    aligned = cv2.imread('static/img_aligned/in_01.png')
+    return cv2.cvtColor(result, cv2.COLOR_RGB2BGR), cv2.cvtColor(aligned, cv2.COLOR_RGB2BGR)
+
+# 创建 Gradio 接口
+style_options = {
+    "Emotion": {'Angry': 'angry', 'Surprised': 'surprised'},
+    "Celebrity": {'Beyonce': 'Beyonce', 'Hilary Clinton': 'Hilary_clinton', 'Johnny Depp': 'Jhonny Depp', 'Taylor Swift': 'Taylor Swift', 'Trump': 'trump'},
+    "Hair Style": {'Afro': 'afro', 'Bowlcut': 'bowlcut', 'Curly Hair': 'curly hair', 'Purple Hair': 'purple hair'}
+}
+
+# 定义更新风格选项的函数
+def update_styles(style_type):
+    if not style_type:
+        return gr.Dropdown(choices=[])
+    return gr.Dropdown(choices=list(style_options[style_type].keys()))
+
+# 创建 Gradio 界面
+with gr.Blocks() as demo:
+    gr.Markdown("# Image Style Transfer")
+    gr.Markdown("### This app is based on styleclip. Choose a style type and a style from the dropdowns below.")
+    with gr.Row():
+        with gr.Column():
+            image_input = gr.Image(type="numpy", label="Upload Image")
+            style_type_dropdown = gr.Dropdown(choices=list(style_options.keys()), label="Style Type")
+            style_dropdown = gr.Dropdown(choices=["Angry", "Curly Hair", "Taylor Swift"], label="Style")
+            style_type_dropdown.change(fn=update_styles, inputs=style_type_dropdown, outputs=style_dropdown)
+            with gr.Row():
+                clear_button = gr.Button("Clear")
+                submit_button = gr.Button("Submit")
+        with gr.Column():
+            output_image = gr.Image(type="numpy", label="Result Image")
+            aligned_image = gr.Image(type="numpy", label="Aligned Image")
+    
+    def on_submit(image, style_type, style):
+        style_value = style_options[style_type][style]
+        return final(image, style_value)
+    
+    def on_clear():
+        return None, None, None, None, None
+    
+    clear_button.click(fn=on_clear, inputs=[], outputs=[image_input, style_type_dropdown, style_dropdown, output_image, aligned_image])
+    submit_button.click(fn=on_submit, inputs=[image_input, style_type_dropdown, style_dropdown], outputs=[output_image, aligned_image])
+
+    examples = gr.Examples(
+        examples=[
+            ["static/img/example1.jpg", "Emotion", "Angry"],
+            ["static/img/example2.jpg", "Celebrity", "Taylor Swift"],
+            ["static/img/example3.jpg", "Hair Style", "Curly Hair"],
+        ],
+        inputs=[image_input, style_type_dropdown, style_dropdown],
+    )
+# 启动 Gradio 应用
+demo.launch()