"O Fascinante Mundo dos Óculos: Estilo, Proteção e Tendências"

app.py

@@ -37,16 +37,16 @@ for idx, name in enumerate(files):
 ep = "assets/examples"
 # Layout, Style, Color, Subject, Prompt，Strict Layout Edge，Layout Content Scale, Automatic Image Matting
 image_examples = [
-    [0, f"{ep}/00_layout.png", f"{ep}/empty.png", f"{ep}/empty.png", f"{ep}/empty.png", "", True, 0., True],
-    [1, f"{ep}/01_layout.png", f"{ep}/empty.png", f"{ep}/empty.png", f"{ep}/empty.png", "", True, 0.8, True],
-    [2, f"{ep}/empty.png", f"{ep}/02_style.png", f"{ep}/empty.png", f"{ep}/empty.png", "", True, 0.8, True],
-    [3, f"{ep}/empty.png", f"{ep}/03_style.png", f"{ep}/empty.png", f"{ep}/empty.png", "", True, 0.8, True],
-    [4, f"{ep}/empty.png", f"{ep}/empty.png", f"{ep}/04_color.png", f"{ep}/empty.png", "A moose, Merry Christmas", True, 0.8, True],
+    [0, f"{ep}/00_layout.png", f"{ep}/empty.png", f"{ep}/empty.png", f"{ep}/empty.png", "", True, 0., False],
+    [1, f"{ep}/01_layout.png", f"{ep}/empty.png", f"{ep}/empty.png", f"{ep}/empty.png", "", True, 0.8, False],
+    [2, f"{ep}/empty.png", f"{ep}/02_style.png", f"{ep}/empty.png", f"{ep}/empty.png", "", True, 0.8, False],
+    [3, f"{ep}/empty.png", f"{ep}/03_style.png", f"{ep}/empty.png", f"{ep}/empty.png", "", True, 0.8, False],
+    [4, f"{ep}/empty.png", f"{ep}/empty.png", f"{ep}/04_color.png", f"{ep}/empty.png", "A moose, Merry Christmas", True, 0.8, False],
     [5, f"{ep}/empty.png", f"{ep}/empty.png", f"{ep}/05_color.png", f"{ep}/empty.png", "A photo about cherry blossom",
-     True, 0.8, True],
-    [6, f"{ep}/06_layout.png", f"{ep}/06_style.jpeg", f"{ep}/empty.png", f"{ep}/empty.png", "", True, 0.8, True],
-    [7, f"{ep}/07_layout.jpeg", f"{ep}/07_style.jpeg", f"{ep}/empty.png", f"{ep}/empty.png", "", True, 0.8, True],
-    [8, f"{ep}/empty.png", f"{ep}/08_style.png", f"{ep}/08_color.jpeg", f"{ep}/empty.png", "", True, 0.8, True],
+     True, 0.8, False],
+    [6, f"{ep}/06_layout.png", f"{ep}/06_style.jpeg", f"{ep}/empty.png", f"{ep}/empty.png", "", True, 0.8, False],
+    [7, f"{ep}/07_layout.jpeg", f"{ep}/07_style.jpeg", f"{ep}/empty.png", f"{ep}/empty.png", "", True, 0.8, False],
+    [8, f"{ep}/empty.png", f"{ep}/08_style.png", f"{ep}/08_color.jpeg", f"{ep}/empty.png", "", True, 0.8, False],
     [9, f"{ep}/empty.png", f"{ep}/09_style.png", f"{ep}/empty.png", f"{ep}/base_image1.jpeg", "", True, 0.8, True],
     [10, f"{ep}/10_layout.png", f"{ep}/10_style.png", f"{ep}/empty.png", f"{ep}/base_image2.png", "", True, 0.8, False],
     [11, f"{ep}/11_layout.png", f"{ep}/empty.png", f"{ep}/empty.png", f"{ep}/base_image4.png", "", False, 0.8, False],
@@ -243,7 +243,7 @@ if __name__ == "__main__":
                         gr.Markdown("""
                         - ⭐️ <b>step1：</b> (Optional) Upload or select a set of images from the examples for the "Layout", "Style", and "Color" reference. Mix and match freely, no need to select all.
                         - ⭐️ <b>step2：</b> (Optional) Use brush to erase areas in the layout, style, and color reference images (if present) that you do not want transferred.
-                        - ⭐️ <b>step3：</b> (Optional) Upload an RGBA image to the "Subject" tab, with the alpha channel indicating the subject you wish to preserve at the pixel level (or upload a regular RGB image and check the automatic image matting option, which will automatically segment the subject for you; <b>the default is the latter</b>).
+                        - ⭐️ <b>step3：</b> (Optional) Upload an image with the subject you wish to preserve at the pixel level to the "Subject" component, with a transparent channel required (or upload a regular RGB image and select automatic image matting, which will automatically segment the subject area for you).
                         - ⭐️ <b>step4：</b> Click "Run" to start the generation process.
                         - ⭐️ <b>step5：</b> (Optional) Additionally, prompt input is supported, as well as control over advanced parameters such as layout edge consistency and conditional weights. Feel free to try these features.
                         """)
@@ -305,7 +305,7 @@ if __name__ == "__main__":
                                                                type="pil", image_mode="RGBA", tool="editor")
                                 pil_base_image_rgba_mirror = gr.Image(label="Subject", image_mode="RGBA", visible=False)
                                 with gr.Box():
-                                    preprocess_base_image = gr.Checkbox(label="Automatic Image Matting", value=True)
+                                    preprocess_base_image = gr.Checkbox(label="Automatic Image Matting")
                                     pil_fg_mask = gr.Image(interactive=False, type="pil", image_mode="L", visible=False)
                         run_button = gr.Button("Run", elem_id="btn")
                         with gr.Accordion("", open=True, elem_id="accordion1"):

requirements.txt

@@ -1,8 +1,5 @@
-sortedcontainers
-addict
-simplejson
+modelscope
 dashscope
-datasets==2.18.0
 tensorflow-gpu==2.11.0
 Pillow
 gradio==3.50.2
@@ -10,4 +7,3 @@ opencv-python
 oss2==2.17.0
 torch
 torchvision
-modelscope

src/util.py

@@ -1,7 +1,6 @@
 import concurrent.futures
 import io
 import os
-import time
 
 import numpy as np
 import oss2
@@ -20,7 +19,7 @@ bucket = oss2.Bucket(oss2.Auth(access_key_id, access_key_secret), endpoint, buck
 oss_path = "hejunjie.hjj/TransferAnythingHF"
 oss_path_img_gallery = "hejunjie.hjj/TransferAnythingHF_img_gallery"
 
-    
+
 def download_img_pil(index, img_url):
     # print(img_url)
     r = requests.get(img_url, stream=True)
@@ -63,10 +62,8 @@ def upload_np_2_oss(input_image, name="cache.png", gallery=False):
     else:
         path = oss_path
 
-    start_time = time.perf_counter()
     bucket.put_object(path + "/" + name, imgByteArr)  # data为数据，可以是图片
     ret = bucket.sign_url('GET', path + "/" + name, 60 * 60 * 24)  # 返回值为链接，参数依次为，方法/oss上文件路径/过期时间(s)
-    logger.info(f"upload cost: {time.perf_counter() - start_time} s.")
     del imgByteArr
     return ret