Spaces:
Sleeping
Sleeping
chengwang96 commited on
Commit ·
3f31363
1
Parent(s): 0ad9500
start with sample
Browse files
app.py
CHANGED
|
@@ -8,15 +8,18 @@ ENHANCE_INPUT_DIR = "sr_input"
|
|
| 8 |
ENHANCE_OUTPUT_DIR = "sr_output"
|
| 9 |
ENHANCE_IMG_NAMES = [f"img_{i}.jpg" for i in range(1, 30)]
|
| 10 |
ENHANCE_CHOICES = ["请选择样例图片或者上传"] + ENHANCE_IMG_NAMES
|
|
|
|
| 11 |
|
| 12 |
SEG_INPUT_DIR = "seg_input"
|
| 13 |
SEG_OUTPUT_DIR = "seg_output"
|
| 14 |
SEG_IMG_NAMES = [f"img_{i}.png" for i in range(1, 30)]
|
| 15 |
SEG_CHOICES = ["请选择样例图片或者上传"] + SEG_IMG_NAMES
|
|
|
|
| 16 |
|
| 17 |
DIAG_INPUT_DIR = "diag_input"
|
| 18 |
DIAG_IMG_NAMES = [f"img_{i}.jpg" for i in range(1, 32)]
|
| 19 |
DIAG_CHOICES = ["请选择样例图片或者上传"] + DIAG_IMG_NAMES
|
|
|
|
| 20 |
|
| 21 |
KVASIR_CLASSES = [
|
| 22 |
"正常粘膜 (normal-z-line)",
|
|
@@ -252,7 +255,7 @@ with gr.Blocks(title="消化道疾病智能分析系统") as demo:
|
|
| 252 |
gr.HTML("<div class='big-title'>图像增强模块</div>")
|
| 253 |
enhance_select = gr.Dropdown(
|
| 254 |
choices=ENHANCE_CHOICES,
|
| 255 |
-
value=
|
| 256 |
label="请选择样例图片或者上传",
|
| 257 |
filterable=True
|
| 258 |
)
|
|
@@ -299,7 +302,7 @@ with gr.Blocks(title="消化道疾病智能分析系统") as demo:
|
|
| 299 |
gr.HTML("<div class='big-title'>图像分割模块</div>")
|
| 300 |
seg_select = gr.Dropdown(
|
| 301 |
choices=SEG_CHOICES,
|
| 302 |
-
value=
|
| 303 |
label="请选择样例图片或者上传",
|
| 304 |
filterable=True
|
| 305 |
)
|
|
@@ -337,7 +340,7 @@ with gr.Blocks(title="消化道疾病智能分析系统") as demo:
|
|
| 337 |
with gr.Column():
|
| 338 |
diag_select = gr.Dropdown(
|
| 339 |
choices=DIAG_CHOICES,
|
| 340 |
-
value=
|
| 341 |
label="请选择样例图片或者上传",
|
| 342 |
filterable=True
|
| 343 |
)
|
|
@@ -374,4 +377,12 @@ with gr.Blocks(title="消化道疾病智能分析系统") as demo:
|
|
| 374 |
outputs=[diag_select, diag_input_img, diag_table, diag_result, diag_suggestion]
|
| 375 |
)
|
| 376 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 377 |
demo.launch()
|
|
|
|
| 8 |
ENHANCE_OUTPUT_DIR = "sr_output"
|
| 9 |
ENHANCE_IMG_NAMES = [f"img_{i}.jpg" for i in range(1, 30)]
|
| 10 |
ENHANCE_CHOICES = ["请选择样例图片或者上传"] + ENHANCE_IMG_NAMES
|
| 11 |
+
ENHANCE_DEFAULT = ENHANCE_IMG_NAMES[0]
|
| 12 |
|
| 13 |
SEG_INPUT_DIR = "seg_input"
|
| 14 |
SEG_OUTPUT_DIR = "seg_output"
|
| 15 |
SEG_IMG_NAMES = [f"img_{i}.png" for i in range(1, 30)]
|
| 16 |
SEG_CHOICES = ["请选择样例图片或者上传"] + SEG_IMG_NAMES
|
| 17 |
+
SEG_DEFAULT = SEG_IMG_NAMES[0]
|
| 18 |
|
| 19 |
DIAG_INPUT_DIR = "diag_input"
|
| 20 |
DIAG_IMG_NAMES = [f"img_{i}.jpg" for i in range(1, 32)]
|
| 21 |
DIAG_CHOICES = ["请选择样例图片或者上传"] + DIAG_IMG_NAMES
|
| 22 |
+
DIAG_DEFAULT = DIAG_IMG_NAMES[0]
|
| 23 |
|
| 24 |
KVASIR_CLASSES = [
|
| 25 |
"正常粘膜 (normal-z-line)",
|
|
|
|
| 255 |
gr.HTML("<div class='big-title'>图像增强模块</div>")
|
| 256 |
enhance_select = gr.Dropdown(
|
| 257 |
choices=ENHANCE_CHOICES,
|
| 258 |
+
value=ENHANCE_DEFAULT,
|
| 259 |
label="请选择样例图片或者上传",
|
| 260 |
filterable=True
|
| 261 |
)
|
|
|
|
| 302 |
gr.HTML("<div class='big-title'>图像分割模块</div>")
|
| 303 |
seg_select = gr.Dropdown(
|
| 304 |
choices=SEG_CHOICES,
|
| 305 |
+
value=SEG_DEFAULT,
|
| 306 |
label="请选择样例图片或者上传",
|
| 307 |
filterable=True
|
| 308 |
)
|
|
|
|
| 340 |
with gr.Column():
|
| 341 |
diag_select = gr.Dropdown(
|
| 342 |
choices=DIAG_CHOICES,
|
| 343 |
+
value=DIAG_DEFAULT,
|
| 344 |
label="请选择样例图片或者上传",
|
| 345 |
filterable=True
|
| 346 |
)
|
|
|
|
| 377 |
outputs=[diag_select, diag_input_img, diag_table, diag_result, diag_suggestion]
|
| 378 |
)
|
| 379 |
|
| 380 |
+
# ====== 页面加载时自动显示默认结果 ======
|
| 381 |
+
demo.load(fn=show_enhance_input, inputs=gr.State(ENHANCE_DEFAULT), outputs=enhance_input_img)
|
| 382 |
+
demo.load(fn=enhance_demo, inputs=gr.State(ENHANCE_DEFAULT), outputs=enhance_output_img)
|
| 383 |
+
demo.load(fn=show_seg_input, inputs=gr.State(SEG_DEFAULT), outputs=seg_input_img)
|
| 384 |
+
demo.load(fn=segment_demo, inputs=gr.State(SEG_DEFAULT), outputs=seg_output_img)
|
| 385 |
+
demo.load(fn=show_diag_input, inputs=gr.State(DIAG_DEFAULT), outputs=diag_input_img)
|
| 386 |
+
demo.load(fn=diagnose_demo, inputs=gr.State(DIAG_DEFAULT), outputs=[diag_table, diag_result, diag_suggestion])
|
| 387 |
+
|
| 388 |
demo.launch()
|