Hugging Face's logo

Spaces:
TencentARC
/
IC-Custom
Running on Zero

App Files Community
1
IC-Custom
File size: 6,354 Bytes 
12edc27
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
 
from PIL import Image

make_dict = lambda x : { 'background': Image.open(x).convert("RGBA"), 'layers': [Image.new("RGBA", Image.open(x).size, (255, 255, 255, 0))], 'composite': Image.open(x).convert("RGBA") }
    
null_dict = {
    'background': None,
    'composite': None,
    'layers': []
}

IMG_REF = [
    ## pos-aware: precise mask
    "assets/gradio/pos_aware/001/img_ref.png",
    "assets/gradio/pos_aware/002/img_ref.png", 
    ## pos-aware: User-drawn mask mask
    "assets/gradio/pos_aware/003/img_ref.png",
    "assets/gradio/pos_aware/004/img_ref.png",
    "assets/gradio/pos_aware/005/img_ref.png",
    ## pos-free
    "assets/gradio/pos_free/001/img_ref.png",
    "assets/gradio/pos_free/002/img_ref.png",
    "assets/gradio/pos_free/003/img_ref.png",
    "assets/gradio/pos_free/004/img_ref.png",
    ]

IMG_TGT1 = [
    ## pos-aware: precise mask
    "assets/gradio/pos_aware/001/img_target.png", 
    "assets/gradio/pos_aware/002/img_target.png", 
    ## pos-aware: User-drawn mask mask
    None,
    None,
    None,
    ## pos-free
    "assets/gradio/pos_free/001/img_target.png", 
    "assets/gradio/pos_free/002/img_target.png",
    "assets/gradio/pos_free/003/img_target.png",
    "assets/gradio/pos_free/004/img_target.png",
    ]

IMG_TGT2 = [
    ## pos-aware: precise mask
    null_dict, 
    null_dict, 
    ## pos-aware: User-drawn mask mask
    make_dict("assets/gradio/pos_aware/003/img_target.png"), 
    make_dict("assets/gradio/pos_aware/004/img_target.png"),
    make_dict("assets/gradio/pos_aware/005/img_target.png"),
    ## pos-free
    null_dict,
    null_dict,
    null_dict,
    null_dict,
    ]

MASK_TGT = [
    ## pos-aware: precise mask
    "assets/gradio/pos_aware/001/mask_target.png", 
    "assets/gradio/pos_aware/002/mask_target.png", 
    ## pos-aware: User-drawn mask mask
    "assets/gradio/pos_aware/003/mask_target.png", 
    "assets/gradio/pos_aware/004/mask_target.png",
    "assets/gradio/pos_aware/005/mask_target.png",
    ## pos-free
    "assets/gradio/pos_free/001/mask_target.png",
    "assets/gradio/pos_free/002/mask_target.png", 
    "assets/gradio/pos_free/003/mask_target.png",
    "assets/gradio/pos_free/004/mask_target.png",
    ]

CUSTOM_MODE = [
    ## pos-aware
    "Position-aware",  
    "Position-aware", 
    "Position-aware", 
    "Position-aware",
    "Position-aware",
    ## pos-free
    "Position-free",
    "Position-free",
    "Position-free",
    "Position-free",
    ]

INPUT_MASK_MODE = [
    ## pos-aware: precise mask
    "Precise mask", 
    "Precise mask", 
    ## pos-aware: User-drawn mask mask
    "User-drawn mask", 
    "User-drawn mask",
    "User-drawn mask",
    ## pos-free
    "Precise mask",
    "Precise mask",
    "Precise mask",
    "Precise mask",
    ]

SEG_REF_MODE = [
    ## pos-aware
    "Full Ref", 
    "Full Ref", 
    "Full Ref", 
    "Full Ref", 
    "Full Ref",
    ## pos-free
    "Full Ref",
    "Full Ref",
    "Full Ref",
    "Full Ref",
    ]

PROMPTS = [
    ## pos-aware: precise mask
    "", 
    "", 
    ## pos-aware: User-drawn mask mask
    "A delicate necklace with a mother-of-pearl clover pendant hangs gracefully around the neck of a woman dressed in a black pinstripe blazer.",
    "",
    "",
    ## pos-free
    "TThe charming, soft plush toy is joyfully wandering through a lush, dense jungle, surrounded by vibrant green foliage and towering trees.",
    "A bright yellow alarm clock sits on a wooden desk next to a stack of books in a cozy, sunlit room.",
    "A Lego figure dressed in a vibrant chicken costume, leaning against a wooden chair, surrounded by lush green grass and blooming flowers.",
    "The crocheted gingerbread man is perched on a tree branch in a dense forest, with sunlight filtering through the leaves, casting dappled shadows around him."
    ]

IMG_GEN = [
    ## pos-aware: precise mask
    "assets/gradio/pos_aware/001/img_gen.png", 
    "assets/gradio/pos_aware/002/img_gen.png", 
    ## pos-aware: User-drawn mask mask
    "assets/gradio/pos_aware/003/img_gen.png",
    "assets/gradio/pos_aware/004/img_gen.png",
    "assets/gradio/pos_aware/005/img_gen.png",
    ## pos-free
    "assets/gradio/pos_free/001/img_gen.png",
    "assets/gradio/pos_free/002/img_gen.png",
    "assets/gradio/pos_free/003/img_gen.png",
    "assets/gradio/pos_free/004/img_gen.png",
    ]

SEED = [
    ## pos-aware
    97175498,
    2126677963, 
    346969695,
    1172525388,
    268683460,
    ## pos-free
    2126677963,
    418898253,
    2126677963,
    2126677963
    ]

TRUE_GS = [
    # pos-aware
    1,
    1,
    1,
    1,
    1,
    # pos-free
    3,
    3,
    3,
    3,
]

NUM_STEPS = [
    ## pos-aware
    32,
    32,
    32,
    32,
    32,
    ## pos-free
    20,
    20,
    20,
    20,
]

GUIDANCE = [
    ## pos-aware
    40,
    48,
    40,
    48,
    48,
    ## pos-free
    40,
    40,
    40,
    40,
]

GRADIO_EXAMPLES = [
    [IMG_REF[0], IMG_TGT1[0], IMG_TGT2[0], CUSTOM_MODE[0], INPUT_MASK_MODE[0], SEG_REF_MODE[0], PROMPTS[0], SEED[0], TRUE_GS[0], '0', NUM_STEPS[0], GUIDANCE[0]],
    [IMG_REF[1], IMG_TGT1[1], IMG_TGT2[1], CUSTOM_MODE[1], INPUT_MASK_MODE[1], SEG_REF_MODE[1], PROMPTS[1], SEED[1], TRUE_GS[1], '1', NUM_STEPS[1], GUIDANCE[1]],
    [IMG_REF[2], IMG_TGT1[2], IMG_TGT2[2], CUSTOM_MODE[2], INPUT_MASK_MODE[2], SEG_REF_MODE[2], PROMPTS[2], SEED[2], TRUE_GS[2], '2', NUM_STEPS[2], GUIDANCE[2]],
    [IMG_REF[3], IMG_TGT1[3], IMG_TGT2[3], CUSTOM_MODE[3], INPUT_MASK_MODE[3], SEG_REF_MODE[3], PROMPTS[3], SEED[3], TRUE_GS[3], '3', NUM_STEPS[3], GUIDANCE[3]],
    [IMG_REF[4], IMG_TGT1[4], IMG_TGT2[4], CUSTOM_MODE[4], INPUT_MASK_MODE[4], SEG_REF_MODE[4], PROMPTS[4], SEED[4], TRUE_GS[4], '4', NUM_STEPS[4], GUIDANCE[4]],
    [IMG_REF[5], IMG_TGT1[5], IMG_TGT2[5], CUSTOM_MODE[5], INPUT_MASK_MODE[5], SEG_REF_MODE[5], PROMPTS[5], SEED[5], TRUE_GS[5], '5', NUM_STEPS[5], GUIDANCE[5]],
    [IMG_REF[6], IMG_TGT1[6], IMG_TGT2[6], CUSTOM_MODE[6], INPUT_MASK_MODE[6], SEG_REF_MODE[6], PROMPTS[6], SEED[6], TRUE_GS[6], '6', NUM_STEPS[6], GUIDANCE[6]],
    [IMG_REF[7], IMG_TGT1[7], IMG_TGT2[7], CUSTOM_MODE[7], INPUT_MASK_MODE[7], SEG_REF_MODE[7], PROMPTS[7], SEED[7], TRUE_GS[7], '7', NUM_STEPS[7], GUIDANCE[7]],
    [IMG_REF[8], IMG_TGT1[8], IMG_TGT2[8], CUSTOM_MODE[8], INPUT_MASK_MODE[8], SEG_REF_MODE[8], PROMPTS[8], SEED[8], TRUE_GS[8], '8', NUM_STEPS[8], GUIDANCE[8]],
]