ameerazam08 commited on
Commit
5147734
1 Parent(s): 53a292b

Upload folder using huggingface_hub

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. README.md +13 -0
  2. app.py +191 -0
  3. examples/basemodel.png +0 -0
  4. examples/garment1.jpg +0 -0
  5. examples/garment1.png +0 -0
  6. examples/garment2.jpg +0 -0
  7. examples/garment2.png +0 -0
  8. examples/garment3.png +0 -0
  9. examples/result1.png +0 -0
  10. examples/result2.png +0 -0
  11. examples/result3.png +0 -0
  12. garments/bottom1.png +0 -0
  13. garments/bottom2.PNG +0 -0
  14. garments/bottom3.JPG +0 -0
  15. garments/bottom4.PNG +0 -0
  16. garments/bottom5.png +0 -0
  17. garments/dress1.png +0 -0
  18. garments/dress2.png +0 -0
  19. garments/top111.png +0 -0
  20. garments/top222.JPG +0 -0
  21. garments/top3.JPG +0 -0
  22. garments/top333.png +0 -0
  23. garments/top4.png +0 -0
  24. garments/top5.png +0 -0
  25. models/eva/Eva_0.png +0 -0
  26. models/eva/Eva_1.png +0 -0
  27. models/rouyan_new/0/densepose.png +0 -0
  28. models/rouyan_new/0/humanmask.png +0 -0
  29. models/rouyan_new/0/initmask.png +0 -0
  30. models/rouyan_new/0/model.png +0 -0
  31. models/rouyan_new/1/densepose.png +0 -0
  32. models/rouyan_new/1/humanmask.png +0 -0
  33. models/rouyan_new/1/initmask.png +0 -0
  34. models/rouyan_new/1/model.png +0 -0
  35. models/rouyan_new/2/densepose.png +0 -0
  36. models/rouyan_new/2/humanmask.png +0 -0
  37. models/rouyan_new/2/initmask.png +0 -0
  38. models/rouyan_new/2/model.png +0 -0
  39. models/rouyan_new/Rouyan_0.png +0 -0
  40. models/rouyan_new/Rouyan_1.png +0 -0
  41. models/rouyan_new/Rouyan_2.png +0 -0
  42. models/simon_online/Simon_0.png +0 -0
  43. models/simon_online/Simon_1.png +0 -0
  44. models/xiaoxuan/model.png +0 -0
  45. models/xiaoxuan_online/Xuanxuan_0.png +0 -0
  46. models/xiaoxuan_online/Xuanxuan_1.png +0 -0
  47. models/xiaoxuan_online/Xuanxuan_2.png +0 -0
  48. models/yaqi/Yaqi_0.png +0 -0
  49. models/yaqi/Yaqi_1.png +0 -0
  50. models/yifeng/model.png +0 -0
README.md ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ title: OutfitAnyone
3
+ emoji: 🏢
4
+ colorFrom: red
5
+ colorTo: pink
6
+ sdk: gradio
7
+ sdk_version: 4.8.0
8
+ app_file: app.py
9
+ pinned: false
10
+ license: cc-by-nc-4.0
11
+ ---
12
+
13
+ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
app.py ADDED
@@ -0,0 +1,191 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import cv2
3
+ import numpy as np
4
+ import json
5
+ import random
6
+ from PIL import Image, ImageDraw, ImageFont
7
+ import asyncio
8
+
9
+ import requests
10
+ import base64
11
+ import gradio as gr
12
+ # from IPython import embed
13
+
14
+ machine_number = 0
15
+ model = os.path.join(os.path.dirname(__file__), "models/eva/Eva_0.png")
16
+
17
+ MODEL_MAP = {
18
+ "AI Model Rouyan_0": 'models/rouyan_new/Rouyan_0.png',
19
+ "AI Model Rouyan_1": 'models/rouyan_new/Rouyan_1.png',
20
+ "AI Model Rouyan_2": 'models/rouyan_new/Rouyan_2.png',
21
+ "AI Model Eva_0": 'models/eva/Eva_0.png',
22
+ "AI Model Eva_1": 'models/eva/Eva_1.png',
23
+ "AI Model Simon_0": 'models/simon_online/Simon_0.png',
24
+ "AI Model Simon_1": 'models/simon_online/Simon_1.png',
25
+ "AI Model Xuanxuan_0": 'models/xiaoxuan_online/Xuanxuan_0.png',
26
+ "AI Model Xuanxuan_1": 'models/xiaoxuan_online/Xuanxuan_1.png',
27
+ "AI Model Xuanxuan_2": 'models/xiaoxuan_online/Xuanxuan_2.png',
28
+ "AI Model Yaqi_0": 'models/yaqi/Yaqi_0.png',
29
+ "AI Model Yaqi_1": 'models/yaqi/Yaqi_1.png',
30
+ "AI Model Yaqi_2": 'models/yaqi/Yaqi_2.png',
31
+ "AI Model Yaqi_3": 'models/yaqi/Yaqi_3.png',
32
+ "AI Model Yifeng_0": 'models/yifeng_online/Yifeng_0.png',
33
+ "AI Model Yifeng_1": 'models/yifeng_online/Yifeng_1.png',
34
+ "AI Model Yifeng_2": 'models/yifeng_online/Yifeng_2.png',
35
+ "AI Model Yifeng_3": 'models/yifeng_online/Yifeng_3.png',
36
+ }
37
+
38
+ def add_waterprint(img):
39
+
40
+ h, w, _ = img.shape
41
+ img = cv2.putText(img, 'Powered by OutfitAnyone', (int(0.3*w), h-20), cv2.FONT_HERSHEY_PLAIN, 2, (128, 128, 128), 2, cv2.LINE_AA)
42
+
43
+ return img
44
+
45
+
46
+ def get_tryon_result(model_name, garment1, garment2, seed=1234):
47
+
48
+ # model_name = "AI Model " + model_name.split("\\")[-1].split(".")[0] # windows
49
+ model_name = "AI Model " + model_name.split("/")[-1].split(".")[0] # linux
50
+ print(model_name)
51
+
52
+ encoded_garment1 = cv2.imencode('.jpg', garment1)[1].tobytes()
53
+ encoded_garment1 = base64.b64encode(encoded_garment1).decode('utf-8')
54
+
55
+ if garment2 is not None:
56
+ encoded_garment2 = cv2.imencode('.jpg', garment2)[1].tobytes()
57
+ encoded_garment2 = base64.b64encode(encoded_garment2).decode('utf-8')
58
+ else:
59
+ encoded_garment2 = ''
60
+
61
+ url = os.environ['OA_IP_ADDRESS']
62
+ headers = {'Content-Type': 'application/json'}
63
+ seed = random.randint(0, 1222222222)
64
+ data = {
65
+ "garment1": encoded_garment1,
66
+ "garment2": encoded_garment2,
67
+ "model_name": model_name,
68
+ "seed": seed
69
+ }
70
+ response = requests.post(url, headers=headers, data=json.dumps(data))
71
+ print("response code", response.status_code)
72
+ if response.status_code == 200:
73
+ result = response.json()
74
+ result = base64.b64decode(result['images'][0])
75
+ result_np = np.frombuffer(result, np.uint8)
76
+ result_img = cv2.imdecode(result_np, cv2.IMREAD_UNCHANGED)
77
+ else:
78
+ print('server error!')
79
+
80
+
81
+ final_img = add_waterprint(result_img)
82
+
83
+ return final_img
84
+
85
+
86
+
87
+ with gr.Blocks(css = ".output-image, .input-image, .image-preview {height: 400px !important} ") as demo:
88
+ # gr.Markdown("# Outfit Anyone v0.9")
89
+ gr.HTML(
90
+ """
91
+ <div style="display: flex; justify-content: center; align-items: center; text-align: center;">
92
+ <a href="https://github.com/HumanAIGC/OutfitAnyone" style="margin-right: 20px; text-decoration: none; display: flex; align-items: center;">
93
+ </a>
94
+ <div>
95
+ <h1 >Outfit Anyone: Ultra-high quality virtual try-on for Any Clothing and Any Person</h1>
96
+ <h4 >v0.9</h4>
97
+ <h5 style="margin: 0;">If you like our project, please give us a star on Github to stay updated with the latest developments.</h5>
98
+ <div style="display: flex; justify-content: center; align-items: center; text-align: center;>
99
+ <a href="https://github.com/HumanAIGC/OutfitAnyone"><img src="https://img.shields.io/badge/Arxiv-0000.00000-red"></a>
100
+ <a href='https://humanaigc.github.io/outfit-anyone/'><img src='https://img.shields.io/badge/Project_Page-OutfitAnyone-green' alt='Project Page'></a>
101
+ <a href='https://github.com/HumanAIGC/OutfitAnyone'><img src='https://img.shields.io/badge/Github-Repo-blue'></a>
102
+ </div>
103
+ </div>
104
+ </div>
105
+ """)
106
+ with gr.Row():
107
+ with gr.Column():
108
+ init_image = gr.Image(sources='clipboard', type="filepath", label="model", value=model)
109
+ example = gr.Examples(inputs=init_image,
110
+ examples_per_page=4,
111
+ examples=[os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Rouyan_0')),
112
+ os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Rouyan_2')),
113
+ os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Eva_0')),
114
+ os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Simon_1')),
115
+ os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Eva_1')),
116
+ os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Simon_0')),
117
+ os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Xuanxuan_0')),
118
+ os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Xuanxuan_2')),
119
+ os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Yaqi_1')),
120
+ os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Yifeng_0')),
121
+ os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Yifeng_3')),
122
+ os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Rouyan_1')),
123
+ os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Yifeng_2')),
124
+ os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Yaqi_0')),
125
+ ])
126
+ with gr.Column():
127
+ gr.HTML(
128
+ """
129
+ <div style="display: flex; justify-content: center; align-items: center; text-align: center;">
130
+ <div>
131
+ <h3>Models are fixed and cannot be uploaded or modified; we only support users uploading their own garments.</h3>
132
+ <h4 style="margin: 0;">For a one-piece dress or coat, you only need to upload the image to the 'top garment' section and leave the 'lower garment' section empty.</h4>
133
+ </div>
134
+ </div>
135
+ """)
136
+ with gr.Row():
137
+ garment_top = gr.Image(sources='upload', type="numpy", label="top garment")
138
+ example_top = gr.Examples(inputs=garment_top,
139
+ examples_per_page=5,
140
+ examples=[os.path.join(os.path.dirname(__file__), "garments/top222.JPG"),
141
+ os.path.join(os.path.dirname(__file__), "garments/top5.png"),
142
+ os.path.join(os.path.dirname(__file__), "garments/top333.png"),
143
+ os.path.join(os.path.dirname(__file__), "garments/dress1.png"),
144
+ os.path.join(os.path.dirname(__file__), "garments/dress2.png"),
145
+ ])
146
+ garment_down = gr.Image(sources='upload', type="numpy", label="lower garment")
147
+ example_down = gr.Examples(inputs=garment_down,
148
+ examples_per_page=5,
149
+ examples=[os.path.join(os.path.dirname(__file__), "garments/bottom1.png"),
150
+ os.path.join(os.path.dirname(__file__), "garments/bottom2.PNG"),
151
+ os.path.join(os.path.dirname(__file__), "garments/bottom3.JPG"),
152
+ os.path.join(os.path.dirname(__file__), "garments/bottom4.PNG"),
153
+ os.path.join(os.path.dirname(__file__), "garments/bottom5.png"),
154
+ ])
155
+
156
+ run_button = gr.Button(value="Run")
157
+ with gr.Column():
158
+ gallery = gr.Image()
159
+
160
+ run_button.click(fn=get_tryon_result,
161
+ inputs=[
162
+ init_image,
163
+ garment_top,
164
+ garment_down,
165
+ ],
166
+ outputs=[gallery],
167
+ concurrency_limit=2)
168
+
169
+
170
+ # Examples
171
+ gr.Markdown("## Examples")
172
+ with gr.Row():
173
+ reference_image1 = gr.Image(label="model", scale=1, value="examples/basemodel.png")
174
+ reference_image2 = gr.Image(label="garment", scale=1, value="examples/garment1.jpg")
175
+ reference_image3 = gr.Image(label="result", scale=1, value="examples/result1.png")
176
+ gr.Examples(
177
+ examples=[
178
+ ["examples/basemodel.png", "examples/garment1.png", "examples/result1.png"],
179
+ ["examples/basemodel.png", "examples/garment2.png", "examples/result2.png"],
180
+ ["examples/basemodel.png", "examples/garment3.png", "examples/result3.png"],
181
+ ],
182
+ inputs=[reference_image1, reference_image2, reference_image3],
183
+ label=None,
184
+ )
185
+
186
+ if __name__ == "__main__":
187
+ ip = requests.get('http://ifconfig.me/ip', timeout=1).text.strip()
188
+ print("ip address alibaba", ip)
189
+ demo.queue(max_size=10)
190
+ demo.launch()
191
+
examples/basemodel.png ADDED
examples/garment1.jpg ADDED
examples/garment1.png ADDED
examples/garment2.jpg ADDED
examples/garment2.png ADDED
examples/garment3.png ADDED
examples/result1.png ADDED
examples/result2.png ADDED
examples/result3.png ADDED
garments/bottom1.png ADDED
garments/bottom2.PNG ADDED
garments/bottom3.JPG ADDED
garments/bottom4.PNG ADDED
garments/bottom5.png ADDED
garments/dress1.png ADDED
garments/dress2.png ADDED
garments/top111.png ADDED
garments/top222.JPG ADDED
garments/top3.JPG ADDED
garments/top333.png ADDED
garments/top4.png ADDED
garments/top5.png ADDED
models/eva/Eva_0.png ADDED
models/eva/Eva_1.png ADDED
models/rouyan_new/0/densepose.png ADDED
models/rouyan_new/0/humanmask.png ADDED
models/rouyan_new/0/initmask.png ADDED
models/rouyan_new/0/model.png ADDED
models/rouyan_new/1/densepose.png ADDED
models/rouyan_new/1/humanmask.png ADDED
models/rouyan_new/1/initmask.png ADDED
models/rouyan_new/1/model.png ADDED
models/rouyan_new/2/densepose.png ADDED
models/rouyan_new/2/humanmask.png ADDED
models/rouyan_new/2/initmask.png ADDED
models/rouyan_new/2/model.png ADDED
models/rouyan_new/Rouyan_0.png ADDED
models/rouyan_new/Rouyan_1.png ADDED
models/rouyan_new/Rouyan_2.png ADDED
models/simon_online/Simon_0.png ADDED
models/simon_online/Simon_1.png ADDED
models/xiaoxuan/model.png ADDED
models/xiaoxuan_online/Xuanxuan_0.png ADDED
models/xiaoxuan_online/Xuanxuan_1.png ADDED
models/xiaoxuan_online/Xuanxuan_2.png ADDED
models/yaqi/Yaqi_0.png ADDED
models/yaqi/Yaqi_1.png ADDED
models/yifeng/model.png ADDED