focuzz commited on
Commit
53771b4
·
verified ·
1 Parent(s): 36ab632

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
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, "Depth (gray)"))
86
- depths_color.append(add_overlay(depth_color, "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="Denoising Steps")
102
- ensemble = gr.Slider(1, 10, value=5, step=1, label="Ensemble Size")
103
- resolution = gr.Slider(256, 1024, value=768, step=64, label="Processing Resolution")
104
- match_res = gr.Checkbox(value=True, label="Match Input Resolution")
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 и Depth", columns=4)
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)