init

.gitattributes

@@ -33,15 +33,6 @@ 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
-Datas/PoseImgs/Camera_XHS_17289683235111040g2sg318bvhdfnk60g49nod0383sqs8u736c0.jpg filter=lfs diff=lfs merge=lfs -text
-Datas/PoseImgs/Camera_XHS_17289683301081040g2sg318bvhdfnk62049nod0383sqsnuu1kng.jpg filter=lfs diff=lfs merge=lfs -text
-Datas/PoseImgs/Camera_XHS_17290497533391040g008318rqal2ck40g435fc4t249gtdfr8tbg.jpg filter=lfs diff=lfs merge=lfs -text
-Datas/PoseImgs/Camera_XHS_17290596474851040g2sg318tukj613q1g40jbin4o5urom8cc8mo.jpg filter=lfs diff=lfs merge=lfs -text
-Datas/PoseImgs/Camera_XHS_17290620188071040g2sg318alvhqvjq7g49nod0383sqsdtdie2g.jpg filter=lfs diff=lfs merge=lfs -text
-Datas/PoseImgs/Camera_XHS_17290620294581040g008318qd9jalgq1g4a0l3adj9gpk5oo3hbg.jpg filter=lfs diff=lfs merge=lfs -text
-Datas/PoseImgs/Camera_XHS_17290690744921040g2sg318b2gi894490414p5j7vc25evc8cj88.jpg filter=lfs diff=lfs merge=lfs -text
-Datas/PoseImgs/Camera_XHS_17292407307731040g2sg318i8mvmv3k0g5nhq88ug8bo472f45lg.jpg filter=lfs diff=lfs merge=lfs -text
-Datas/PoseImgs/Camera_XHS_17292407361051040g2sg31927euo04u7043nr4n5hrue9dvm5n5g.jpg filter=lfs diff=lfs merge=lfs -text
-Datas/PoseImgs/Camera_XHS_17292407854531040g00830okpafop2c0049vv6g4hguaccc6d2ng.jpg filter=lfs diff=lfs merge=lfs -text
-Datas/PoseImgs/Camera_XHS_17292408016631040g008313kg6jdo000049ch9ulbcgkl0hvvp68.jpg filter=lfs diff=lfs merge=lfs -text
-Datas/PoseImgs/Camera_XHS_17292408186161040g2sg30qqb14ohgskg5nqk8ljg80munke428o.jpg filter=lfs diff=lfs merge=lfs -text
+*.png filter=lfs diff=lfs merge=lfs -text
+*.gif filter=lfs diff=lfs merge=lfs -text
+*.jpg filter=lfs diff=lfs merge=lfs -text

.gitignore

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

app.py

@@ -0,0 +1,149 @@
+
+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()
+tip1, tip2 = get_tips()
+face_detector = MTCNN()
+
+# Description
+title = r"""
+<h1 align="center">Outfit Anyway: Best customer try-on You ever See</h1>
+"""
+
+description = r"""
+<b>Join discord to know more about </b> <a href='https://discord.com/invite/QgJWCtSG58' target='_blank'><b> heybeauty prebuy vton solution</b></a>.<br>
+"""
+
+
+def onClick(cloth_image, pose_image, high_resolution, 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", ""
+
+        if high_resolution:
+            public_res = publicClothSwap(upload_url, cloth_id, is_hr=1)
+        else:
+            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, task id: {public_res['id']}, {public_res['msg']}", ""
+
+        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 public_res['mid_result'] + f"?t={timestamp}", "task query failed,", ""
+            elif state['status']=='PROCESSING':
+                yield public_res['mid_result'] + f"?t={timestamp}", f"task is processing, query {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 public_res['mid_result'] + f"?t={timestamp}", f"task is on processing, query {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)
+    gr.Markdown(description)
+
+    with gr.Accordion('upload tips', open=False):
+        with gr.Row():
+            gr.HTML(f"<img src=\"{tip1}\" >")
+            gr.HTML(f"<img src=\"{tip2}\" >")
+                    
+    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():
+                # size_slider = gr.Slider(-3, 3, value=1, interactive=True, label="clothes size")
+                high_resolution = gr.Checkbox(value=False, label="high resolution", interactive=True)
+                
+                run_button = gr.Button(value="Run")
+                info_text = gr.Textbox(value="", interactive=False, 
+                    label='runtime information')                
+                res_image = gr.Image(label="result image", value=None, type="filepath")
+                MK01 = gr.Markdown()
+
+    run_button.click(fn=onClick, inputs=[cloth_image, pose_image, high_resolution], 
+                     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')
+    

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 = "https://selfit-deploy-1256039085.cos.accelerate.myqcloud.com/"
+Regions = "IndiaPakistanBengal"
+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