Spaces:
Runtime error
Runtime error
suppress error on clear image
Browse files
app.py
CHANGED
|
@@ -62,45 +62,53 @@ def subset_to_json_string(arr):
|
|
| 62 |
return '[' + arr_str + ']'
|
| 63 |
|
| 64 |
def estimate_body(source):
|
| 65 |
-
print("estimate_body")
|
| 66 |
if source == None:
|
| 67 |
return None
|
| 68 |
|
| 69 |
candidate, subset = body_estimation(pil2cv(source))
|
| 70 |
-
print(candidate_to_json_string(candidate))
|
| 71 |
-
print(subset_to_json_string(subset))
|
| 72 |
return "{ \"candidate\": " + candidate_to_json_string(candidate) + ", \"subset\": " + subset_to_json_string(subset) + " }"
|
| 73 |
|
| 74 |
def image_changed(image):
|
| 75 |
if (image == None):
|
| 76 |
-
return
|
| 77 |
json = estimate_body(image)
|
| 78 |
return json, image.width, image.height
|
| 79 |
|
| 80 |
-
def image_preprocess(image):
|
| 81 |
-
return image is not None
|
| 82 |
-
|
| 83 |
html_text = f"""
|
| 84 |
<canvas id="canvas" width="512" height="512"></canvas>
|
| 85 |
<script type="text/javascript" defer>{file_contents}</script>
|
| 86 |
"""
|
| 87 |
|
| 88 |
with gr.Blocks() as demo:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 89 |
with gr.Row():
|
| 90 |
with gr.Column(scale=1):
|
| 91 |
source = gr.Image(type="pil")
|
| 92 |
width = gr.Slider(label="Width", mininmum=512, maximum=1024, step=64, value=512, key="Width", interactive=True)
|
| 93 |
height = gr.Slider(label="Height", mininmum=512, maximum=1024, step=64, value=512, key="Height", interactive=True)
|
| 94 |
startBtn = gr.Button(value="Start edit")
|
| 95 |
-
json = gr.JSON(label="
|
|
|
|
|
|
|
| 96 |
with gr.Column(scale=2):
|
| 97 |
-
gr.HTML("<ul><li>ctrl + drag to scale</li><li>alt + drag to translate</li><li>shift + drag to rotate(move right first, then up or down)</li></ul>")
|
| 98 |
html = gr.HTML(html_text)
|
| 99 |
saveBtn = gr.Button(value="Save")
|
|
|
|
| 100 |
|
| 101 |
source.change(
|
| 102 |
fn = image_changed,
|
| 103 |
-
preprocess = image_preprocess,
|
| 104 |
inputs = [source],
|
| 105 |
outputs = [json, width, height])
|
| 106 |
startBtn.click(
|
|
@@ -112,5 +120,8 @@ with gr.Blocks() as demo:
|
|
| 112 |
fn = None,
|
| 113 |
inputs = [], outputs = [json],
|
| 114 |
_js="() => { return [savePose()]; }")
|
| 115 |
-
|
|
|
|
|
|
|
|
|
|
| 116 |
gr.mount_gradio_app(app, demo, path="/")
|
|
|
|
| 62 |
return '[' + arr_str + ']'
|
| 63 |
|
| 64 |
def estimate_body(source):
|
|
|
|
| 65 |
if source == None:
|
| 66 |
return None
|
| 67 |
|
| 68 |
candidate, subset = body_estimation(pil2cv(source))
|
|
|
|
|
|
|
| 69 |
return "{ \"candidate\": " + candidate_to_json_string(candidate) + ", \"subset\": " + subset_to_json_string(subset) + " }"
|
| 70 |
|
| 71 |
def image_changed(image):
|
| 72 |
if (image == None):
|
| 73 |
+
return {}, 512, 512
|
| 74 |
json = estimate_body(image)
|
| 75 |
return json, image.width, image.height
|
| 76 |
|
|
|
|
|
|
|
|
|
|
| 77 |
html_text = f"""
|
| 78 |
<canvas id="canvas" width="512" height="512"></canvas>
|
| 79 |
<script type="text/javascript" defer>{file_contents}</script>
|
| 80 |
"""
|
| 81 |
|
| 82 |
with gr.Blocks() as demo:
|
| 83 |
+
gr.Markdown("""### Usage
|
| 84 |
+
|
| 85 |
+
Choose one of the following methods to edit the pose:
|
| 86 |
+
|
| 87 |
+
| Style | Description |
|
| 88 |
+
| -----------------| ----------------------------------------------------------------------------------------- |
|
| 89 |
+
| Pose recognition | Upload an image and click "Start edit". |
|
| 90 |
+
| Input json | Input json to "Json source" and click "Input Json", edit the width/height, then click "Start edit". |
|
| 91 |
+
| Free style | Edit the width/height, then click "Start edit". |
|
| 92 |
+
|
| 93 |
+
To save the pose image, click "Save".
|
| 94 |
+
To export the pose data, click "Save" and "Copy to clipboard" of "Json" section.
|
| 95 |
+
""")
|
| 96 |
with gr.Row():
|
| 97 |
with gr.Column(scale=1):
|
| 98 |
source = gr.Image(type="pil")
|
| 99 |
width = gr.Slider(label="Width", mininmum=512, maximum=1024, step=64, value=512, key="Width", interactive=True)
|
| 100 |
height = gr.Slider(label="Height", mininmum=512, maximum=1024, step=64, value=512, key="Height", interactive=True)
|
| 101 |
startBtn = gr.Button(value="Start edit")
|
| 102 |
+
json = gr.JSON(label="Json", lines=10)
|
| 103 |
+
jsonInput = gr.Textbox(label="Json source", lines=10)
|
| 104 |
+
jsonInputBtn = gr.Button(value="Input Json")
|
| 105 |
with gr.Column(scale=2):
|
|
|
|
| 106 |
html = gr.HTML(html_text)
|
| 107 |
saveBtn = gr.Button(value="Save")
|
| 108 |
+
gr.HTML("<ul><li>ctrl + drag to scale</li><li>alt + drag to translate</li><li>shift + drag to rotate(move right first, then up or down)</li></ul>")
|
| 109 |
|
| 110 |
source.change(
|
| 111 |
fn = image_changed,
|
|
|
|
| 112 |
inputs = [source],
|
| 113 |
outputs = [json, width, height])
|
| 114 |
startBtn.click(
|
|
|
|
| 120 |
fn = None,
|
| 121 |
inputs = [], outputs = [json],
|
| 122 |
_js="() => { return [savePose()]; }")
|
| 123 |
+
jsonInputBtn.click(
|
| 124 |
+
fn = lambda x: x,
|
| 125 |
+
inputs = [jsonInput], outputs = [json])
|
| 126 |
+
|
| 127 |
gr.mount_gradio_app(app, demo, path="/")
|