Update app.py
Browse files
app.py
CHANGED
|
@@ -82,8 +82,8 @@ def generate_gallery():
|
|
| 82 |
depth_color = output.depth_colored.resize(TARGET_SIZE, Image.BILINEAR)
|
| 83 |
|
| 84 |
rgbs.append(add_overlay(rgb, "RGB"))
|
| 85 |
-
depths_gray.append(add_overlay(depth_gray, "
|
| 86 |
-
depths_color.append(add_overlay(depth_color, "
|
| 87 |
|
| 88 |
return rgbs + depths_color + depths_gray
|
| 89 |
|
|
@@ -98,10 +98,10 @@ with gr.Blocks() as demo:
|
|
| 98 |
with gr.Row():
|
| 99 |
with gr.Column(scale=1):
|
| 100 |
input_image = gr.Image(label="Загрузите RGB изображение", type="pil")
|
| 101 |
-
denoise = gr.Slider(1, 50, value=4, step=1, label="
|
| 102 |
-
ensemble = gr.Slider(1, 10, value=5, step=1, label="
|
| 103 |
-
resolution = gr.Slider(256, 1024, value=768, step=64, label="
|
| 104 |
-
match_res = gr.Checkbox(value=True, label="
|
| 105 |
with gr.Column(scale=1):
|
| 106 |
output_image = gr.Image(label="Карта глубины")
|
| 107 |
|
|
@@ -127,7 +127,7 @@ with gr.Blocks() as demo:
|
|
| 127 |
)
|
| 128 |
|
| 129 |
gr.Markdown("### Примеры:")
|
| 130 |
-
gallery = gr.Gallery(label="Сравнение RGB и
|
| 131 |
demo.load(fn=generate_gallery, outputs=gallery)
|
| 132 |
|
| 133 |
demo.launch(share=True)
|
|
|
|
| 82 |
depth_color = output.depth_colored.resize(TARGET_SIZE, Image.BILINEAR)
|
| 83 |
|
| 84 |
rgbs.append(add_overlay(rgb, "RGB"))
|
| 85 |
+
depths_gray.append(add_overlay(depth_gray, "Глубина (серая)"))
|
| 86 |
+
depths_color.append(add_overlay(depth_color, "Глубина (цветная)"))
|
| 87 |
|
| 88 |
return rgbs + depths_color + depths_gray
|
| 89 |
|
|
|
|
| 98 |
with gr.Row():
|
| 99 |
with gr.Column(scale=1):
|
| 100 |
input_image = gr.Image(label="Загрузите RGB изображение", type="pil")
|
| 101 |
+
denoise = gr.Slider(1, 50, value=4, step=1, label="Шаги денойзинга")
|
| 102 |
+
ensemble = gr.Slider(1, 10, value=5, step=1, label="Размер ансамбля (количество запусков для одной картинки)")
|
| 103 |
+
resolution = gr.Slider(256, 1024, value=768, step=64, label="Разрешение обработки изображений")
|
| 104 |
+
match_res = gr.Checkbox(value=True, label="Сохранять исходное разрешение")
|
| 105 |
with gr.Column(scale=1):
|
| 106 |
output_image = gr.Image(label="Карта глубины")
|
| 107 |
|
|
|
|
| 127 |
)
|
| 128 |
|
| 129 |
gr.Markdown("### Примеры:")
|
| 130 |
+
gallery = gr.Gallery(label="Сравнение RGB и Глубины", columns=4)
|
| 131 |
demo.load(fn=generate_gallery, outputs=gallery)
|
| 132 |
|
| 133 |
demo.launch(share=True)
|