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)
|