Upload 7 files

.gitattributes

@@ -1,35 +1,36 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz 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
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz 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
+*.jpg filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,2 @@
+order.txt
+push.sh

README.md

@@ -1,13 +1,13 @@
----
-title: Vtondemo
-emoji: 🏆
-colorFrom: pink
-colorTo: indigo
-sdk: gradio
-sdk_version: 5.16.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
+---
+title: Vtondemo
+emoji: 🏆
+colorFrom: pink
+colorTo: indigo
+sdk: gradio
+sdk_version: 5.16.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

app.py

@@ -0,0 +1,133 @@
+
+import cv2
+from mtcnn.mtcnn import MTCNN
+from utils import *
+
+
+cloth_examples = get_cloth_examples(hr=0)
+cloth_hr_examples = get_cloth_examples(hr=1)
+pose_examples = get_pose_examples()
+face_detector = MTCNN()
+
+# Description
+title = r"""
+<h1 align="center">VTON Test Demo</h1>
+"""
+
+
+def onClick(cloth_image, pose_image, request: gr.Request):
+    if pose_image is None:
+        yield None, "no pose image found !", ""
+        return None, "no pose image found !", ""
+    if cloth_image is None:
+        yield None, "no cloth image found !", ""
+        return None, "no cloth image found !", ""
+
+    pose_id = os.path.basename(pose_image).split(".")[0]
+    cloth_id = int(os.path.basename(cloth_image).split(".")[0])
+            
+    try:
+
+        client_ip = request.client.host
+        x_forwarded_for = dict(request.headers).get('x-forwarded-for')
+        if x_forwarded_for:
+            client_ip = x_forwarded_for
+            
+        pose_np = cv2.imread(pose_image)
+        faces = face_detector.detect_faces(pose_np[:,:,::-1])
+        if len(faces)==0:
+            print(client_ip, 'faces num is 0! ', flush=True)
+            yield None, "Fatal Error !!! No face detected !!! You must upload a human photo!!! Not clothing photo!!!", ""
+            return None, "Fatal Error !!! No face detected !!! You must upload a human photo!!! Not clothing photo!!!", ""
+        else:
+            x, y, w, h = faces[0]["box"]
+            H, W = pose_np.shape[:2]
+            max_face_ratio = 1/3.3
+            if w/W>max_face_ratio or h/H>max_face_ratio:
+                yield None, "Fatal Error !!! Headshot is not allowed !!! You must upload a full-body or half-body photo!!!", ""
+                return None, "Fatal Error !!! Headshot is not allowed !!! You must upload a full-body or half-body photo!!!", ""
+
+        if not check_region_warp(client_ip):
+            yield None, "Failed !!! Our server is under maintenance, please try again later", ""
+            return None, "Failed !!! Our server is under maintenance, please try again later", ""
+            
+        # client_ip = '8.8.8.8'
+        yield None, "begin to upload ", ""
+
+        timeId = int(  str(time.time()).replace(".", "")  )+random.randint(1000, 9999)
+        upload_url = upload_pose_img(client_ip, timeId, pose_image)
+        # exit(0)
+        yield None, "begin to public task ", ""
+        # return None, "begin to public task ", ""
+        
+        if len(upload_url)==0:
+            yield None, "fail to upload", ""
+            return None, "fail to upload", ""
+
+        public_res = publicClothSwap(upload_url, cloth_id, is_hr=0)
+        if public_res is None:
+            yield None, "fail to public you task", ""
+            return None, "fail to public you task", ""
+
+        print(client_ip, public_res['mid_result'])
+        yield public_res['mid_result'], f"task is processing", ""
+
+        max_try = 120*3
+        wait_s = 0.5
+        for i in range(max_try):
+            time.sleep(wait_s)
+            state = getInfRes(public_res['id'])
+            timestamp = int(time.time() * 1000)
+            if state is None:
+                yield None, "task query failed,", ""
+            elif state['status']=='PROCESSING':
+                yield None, f"wait {i}", ""
+            elif state['status']=='SUCCEED':
+                yield state['output1'] + f"?t={timestamp}", f"task finished, {state['msg']}", ""
+                return state['output1'] + f"?t={timestamp}", f"task finished, {state['msg']}", ""
+            elif state['status']=='FAILED':
+                yield None, f"task failed, {state['msg']}", ""
+                return None, f"task failed, {state['msg']}", ""
+            else:
+                yield None, f"{i}", ""
+        return None, "no machine...", ""
+    except Exception as e:
+        print(e)
+        raise e
+        return None, "fail to create task", ""
+
+with gr.Blocks() as demo:
+    gr.Markdown(title)
+
+    with gr.Row():
+        with gr.Column():
+            cloth_image = gr.Image(value=None, interactive=False, type="filepath", label="choose a clothing")
+            example = gr.Examples(inputs=cloth_image,examples_per_page=20,examples=cloth_examples, label="clothing")
+            hr_example = gr.Examples(inputs=cloth_image,examples_per_page=9,examples=cloth_hr_examples, label="invalid clothing")
+            
+        with gr.Column():
+            pose_image = gr.Image(value=None, type="filepath", label="choose/upload a photo")
+            example_pose = gr.Examples(inputs=pose_image,
+                                      examples_per_page=20,
+                                      examples=pose_examples)
+            
+        with gr.Column():
+            with gr.Column():
+                
+                run_button = gr.Button(value="Submit")
+                info_text = gr.Textbox(value="", interactive=False, 
+                    label='log')                
+                res_image = gr.Image(label="result image", value=None, type="filepath")
+                MK01 = gr.Markdown()
+
+    run_button.click(fn=onClick, inputs=[cloth_image, pose_image], 
+                     outputs=[res_image, info_text, MK01])
+
+
+if __name__ == "__main__":
+
+    demo.queue(max_size=50)
+    # demo.queue(concurrency_count=60)
+    # demo.launch(server_name='0.0.0.0', server_port=225)
+    demo.launch(server_name='0.0.0.0')
+    

push.sh

@@ -0,0 +1,9 @@
+
+
+git config --global http.proxy http://127.0.0.1:7890
+git config --global https.proxy http://127.0.0.1:7890
+
+git add .
+git commit -m "init"
+git push
+

requirements.txt

@@ -0,0 +1,8 @@
+opencv-python
+numpy
+requests
+gradio==4.40.0
+gradio-client==1.2.0
+func_timeout
+mtcnn
+tensorflow

utils.py

@@ -0,0 +1,176 @@
+
+import os
+import sys
+import cv2
+import json
+import random
+import time
+import requests
+import func_timeout
+import numpy as np
+import gradio as gr
+
+
+OssUrl = ""
+Regions = ""
+TOKEN = os.environ['TOKEN']
+UKAPIURL = os.environ['UKAPIURL']
+
+
+proj_dir = os.path.dirname(os.path.abspath(__file__))
+data_dir = os.path.join(proj_dir, 'Datas')
+# data_dir = "Datas"
+tmpFolder = "tmp"
+os.makedirs(tmpFolder, exist_ok=True)
+
+
+def get_cloth_examples(hr=0):
+    cloth_dir = os.path.join(data_dir, 'ClothImgs')
+    examples = []
+    files = sorted(os.listdir(cloth_dir))
+    hr_clothes = list(range(588, 597))
+    # hr_clothes = []
+    for f in files:
+        if '.jpg' not in f and '.png' not in f:
+            continue
+
+        cloth_id = f.split(".")[0]
+        if int(cloth_id) in hr_clothes and hr==0:
+            continue
+        if int(cloth_id) not in hr_clothes and hr==1:
+            continue
+
+        cloth_path = os.path.join(cloth_dir, f)
+        examples.append(cloth_path)
+    examples = examples[::-1]
+    return examples
+
+def get_pose_examples():
+    pose_dir = os.path.join(data_dir, 'PoseImgs')
+    examples = []
+    for f in os.listdir(pose_dir):
+        if '.jpg' not in f and '.png' not in f:
+            continue
+        pose_id = f.split(".")[0]
+        pose_path = os.path.join(pose_dir, f)
+        examples.append(pose_path)
+    return examples
+
+def get_result_example(cloth_id, pose_id):
+    result_dir = os.path.join(data_dir, 'ResultImgs')
+    res_path = os.path.join(result_dir, f"{cloth_id}_{pose_id}.jpg")
+    return res_path
+
+# def get_tips():
+#     path1 = OssUrl+'ClothData/Publics/PoseGuide/tip1.jpg'
+#     path2 = OssUrl+'ClothData/Publics/PoseGuide/tip2.jpg'
+#     return path1, path2
+
+def upload_pose_img(clientIp, timeId, img):
+    fileName = clientIp.replace(".", "")+str(timeId)+".jpg"
+    local_path = os.path.join(tmpFolder, fileName)
+    img = cv2.imread(img)
+    cv2.imwrite(os.path.join(tmpFolder, fileName), img)
+
+    json_data = {
+        "token": "c0e69e5d129b11efa10c525400b75156",  
+        "input1": fileName,
+        "input2": "",
+        "protocol": "",
+        "cloud": "ali"
+    }
+    
+    session = requests.session()
+    ret = requests.post(
+        f"{UKAPIURL}/upload",
+        headers={'Content-Type': 'application/json'},
+        json=json_data
+    )
+    
+    res = ""
+    if ret.status_code==200:
+        if 'upload1' in ret.json():
+            upload_url = ret.json()['upload1']
+            headers = {'Content-Type': 'image/jpeg'}
+            response = session.put(upload_url, data=open(local_path, 'rb').read(), headers=headers)
+            # print(response.status_code)
+            if response.status_code == 200:
+                res = upload_url
+    if os.path.exists(local_path):
+        os.remove(local_path)
+    return res
+
+
+def publicClothSwap(image, clothId, is_hr=0):
+    json_data = {
+        "image": image,
+        "task_type": "11", 
+        "param1": str(clothId),
+        "param2": "",
+        "param3": "",
+        "param4": str(is_hr),
+        "delete_if_complete": "1",
+        "force_celery":"1"
+    }
+    
+    headers = {
+        'Authorization': f'Bearer {TOKEN}',
+        'Content-Type': 'application/json'
+    }
+    
+    ret = requests.post(
+        f'{UKAPIURL}/public_advton',
+        headers=headers,
+        json=json_data
+    )
+    
+    if ret.status_code == 200:
+        response = ret.json()
+        if 'mid_result' in response and 'id' in response:
+            return {'mid_result': response['mid_result'], 'id': response['id'], "msg": response['msg']}
+        # print(response)
+        return None
+    else:
+        return None
+
+def getInfRes(taskId):
+    headers = {
+        'Content-Type': 'application/json'
+    }
+    json_data = {
+        'id': taskId
+    }
+    ret = requests.post(
+        f'{UKAPIURL}/status_advton',
+        headers=headers,
+        json=json_data
+    )
+    if ret.status_code == 200:
+        response = ret.json()
+        if 'status' in response:
+            return response
+        print(response)
+        return None
+    else:
+        return None
+
+@func_timeout.func_set_timeout(10)
+def check_region(ip):
+    session = requests.session()
+    # ret = requests.get(f"https://api.ip2location.io/?ip={ip}")
+    ret = requests.get(f"https://realip.cc/?ip={ip}")
+    # print(ret)
+    nat = ret.json()['country'].lower()
+    if nat in Regions.lower():
+        print(nat, 'invalid', ip)
+        return False
+    else:
+        print(nat, 'valid', ip)
+    return True
+
+def check_region_warp(ip):
+    try:
+        return check_region(ip)
+    except Exception as e:
+        print(e)
+        return True