diff --git a/app.py b/app.py index 1cf3051f18b40ee9b474b16d5f4e43fe89a15611..6a77b0d19518d4a1a4fe537740138d2c3a13d0ac 100644 --- a/app.py +++ b/app.py @@ -5,18 +5,18 @@ import random # ========== 配置图片路径 ========== # 图像增强 -ENHANCE_INPUT_DIR = "f1_input" -ENHANCE_OUTPUT_DIR = "f1_output" -ENHANCE_IMG_NAMES = [f"{i}.png" for i in range(1, 613)] # 1.png ~ 612.png +ENHANCE_INPUT_DIR = "sr_input" +ENHANCE_OUTPUT_DIR = "sr_output" +ENHANCE_IMG_NAMES = [f"{i}.png" for i in range(1, 30)] # 分割 SEG_INPUT_DIR = "seg_input" SEG_OUTPUT_DIR = "seg_output" -SEG_IMG_NAMES = [f"{i}.png" for i in range(1, 6)] # 1.png ~ 5.png +SEG_IMG_NAMES = [f"{i}.png" for i in range(1, 30)] # 诊断 DIAG_INPUT_DIR = "diag_input" -DIAG_IMG_NAMES = [f"{i}.png" for i in range(1, 6)] # 1.png ~ 5.png +DIAG_IMG_NAMES = [f"{i}.png" for i in range(1, 30)] # Kvasir 类别 KVASIR_CLASSES = [ @@ -38,7 +38,6 @@ def open_img(img_path): return Image.new("RGB", (256, 256), (200, 200, 200)) return Image.open(img_path) -# 图像增强模块 def show_enhance_input(img_name): path = os.path.join(ENHANCE_INPUT_DIR, img_name) return open_img(path) @@ -47,7 +46,6 @@ def enhance_demo(img_name): path = os.path.join(ENHANCE_OUTPUT_DIR, img_name) return open_img(path) -# 图像分割模块 def show_seg_input(img_name): path = os.path.join(SEG_INPUT_DIR, img_name) return open_img(path) @@ -56,7 +54,6 @@ def segment_demo(img_name): path = os.path.join(SEG_OUTPUT_DIR, img_name) return open_img(path) -# 疾病诊断模块 def show_diag_input(img_name): path = os.path.join(DIAG_INPUT_DIR, img_name) return open_img(path) @@ -70,14 +67,15 @@ def diagnose_demo(img_name): with gr.Blocks(title="AI内窥镜影像辅助分析系统") as demo: gr.Markdown("## AI内窥镜影像辅助分析系统") - # 上方：图像增强模块 + # 上方：图像增强模块（输入和输出在一行） with gr.Row(): with gr.Column(): gr.Markdown("### 图像增强模块") enhance_select = gr.Dropdown(choices=ENHANCE_IMG_NAMES, value=ENHANCE_IMG_NAMES[0], label="选择样例图片", filterable=True) - enhance_input_img = gr.Image(label="原始图片", interactive=False) + with gr.Row(): + enhance_input_img = gr.Image(label="原始图片", interactive=False) + enhance_output_img = gr.Image(label="增强结果图片", interactive=False) enhance_btn = gr.Button("增强图片") - enhance_output_img = gr.Image(label="增强结果图片", interactive=False) # 选择图片自动显示原图 enhance_select.change(fn=show_enhance_input, inputs=enhance_select, outputs=enhance_input_img) diff --git a/f1_input/1.png b/seg_input/1.png similarity index 100% rename from f1_input/1.png rename to seg_input/1.png diff --git a/f1_input/10.png b/seg_input/10.png similarity index 100% rename from f1_input/10.png rename to seg_input/10.png diff --git a/f1_input/11.png b/seg_input/11.png similarity index 100% rename from f1_input/11.png rename to seg_input/11.png diff --git a/f1_input/12.png b/seg_input/12.png similarity index 100% rename from f1_input/12.png rename to seg_input/12.png diff --git a/f1_input/13.png b/seg_input/13.png similarity index 100% rename from f1_input/13.png rename to seg_input/13.png diff --git a/f1_input/14.png b/seg_input/14.png similarity index 100% rename from f1_input/14.png rename to seg_input/14.png diff --git a/f1_input/15.png b/seg_input/15.png similarity index 100% rename from f1_input/15.png rename to seg_input/15.png diff --git a/f1_input/16.png b/seg_input/16.png similarity index 100% rename from f1_input/16.png rename to seg_input/16.png diff --git a/f1_input/17.png b/seg_input/17.png similarity index 100% rename from f1_input/17.png rename to seg_input/17.png diff --git a/f1_input/18.png b/seg_input/18.png similarity index 100% rename from f1_input/18.png rename to seg_input/18.png diff --git a/f1_input/19.png b/seg_input/19.png similarity index 100% rename from f1_input/19.png rename to seg_input/19.png diff --git a/f1_input/2.png b/seg_input/2.png similarity index 100% rename from f1_input/2.png rename to seg_input/2.png diff --git a/f1_input/20.png b/seg_input/20.png similarity index 100% rename from f1_input/20.png rename to seg_input/20.png diff --git a/f1_input/21.png b/seg_input/21.png similarity index 100% rename from f1_input/21.png rename to seg_input/21.png diff --git a/f1_input/22.png b/seg_input/22.png similarity index 100% rename from f1_input/22.png rename to seg_input/22.png diff --git a/f1_input/23.png b/seg_input/23.png similarity index 100% rename from f1_input/23.png rename to seg_input/23.png diff --git a/f1_input/24.png b/seg_input/24.png similarity index 100% rename from f1_input/24.png rename to seg_input/24.png diff --git a/f1_input/25.png b/seg_input/25.png similarity index 100% rename from f1_input/25.png rename to seg_input/25.png diff --git a/f1_input/26.png b/seg_input/26.png similarity index 100% rename from f1_input/26.png rename to seg_input/26.png diff --git a/f1_input/27.png b/seg_input/27.png similarity index 100% rename from f1_input/27.png rename to seg_input/27.png diff --git a/f1_input/28.png b/seg_input/28.png similarity index 100% rename from f1_input/28.png rename to seg_input/28.png diff --git a/f1_input/29.png b/seg_input/29.png similarity index 100% rename from f1_input/29.png rename to seg_input/29.png diff --git a/f1_input/3.png b/seg_input/3.png similarity index 100% rename from f1_input/3.png rename to seg_input/3.png diff --git a/f1_input/30.png b/seg_input/30.png similarity index 100% rename from f1_input/30.png rename to seg_input/30.png diff --git a/f1_input/4.png b/seg_input/4.png similarity index 100% rename from f1_input/4.png rename to seg_input/4.png diff --git a/f1_input/5.png b/seg_input/5.png similarity index 100% rename from f1_input/5.png rename to seg_input/5.png diff --git a/f1_input/6.png b/seg_input/6.png similarity index 100% rename from f1_input/6.png rename to seg_input/6.png diff --git a/f1_input/7.png b/seg_input/7.png similarity index 100% rename from f1_input/7.png rename to seg_input/7.png diff --git a/f1_input/8.png b/seg_input/8.png similarity index 100% rename from f1_input/8.png rename to seg_input/8.png diff --git a/f1_input/9.png b/seg_input/9.png similarity index 100% rename from f1_input/9.png rename to seg_input/9.png diff --git a/f1_output/1.png b/seg_output/1.png similarity index 100% rename from f1_output/1.png rename to seg_output/1.png diff --git a/f1_output/10.png b/seg_output/10.png similarity index 100% rename from f1_output/10.png rename to seg_output/10.png diff --git a/f1_output/11.png b/seg_output/11.png similarity index 100% rename from f1_output/11.png rename to seg_output/11.png diff --git a/f1_output/12.png b/seg_output/12.png similarity index 100% rename from f1_output/12.png rename to seg_output/12.png diff --git a/f1_output/13.png b/seg_output/13.png similarity index 100% rename from f1_output/13.png rename to seg_output/13.png diff --git a/f1_output/14.png b/seg_output/14.png similarity index 100% rename from f1_output/14.png rename to seg_output/14.png diff --git a/f1_output/15.png b/seg_output/15.png similarity index 100% rename from f1_output/15.png rename to seg_output/15.png diff --git a/f1_output/16.png b/seg_output/16.png similarity index 100% rename from f1_output/16.png rename to seg_output/16.png diff --git a/f1_output/17.png b/seg_output/17.png similarity index 100% rename from f1_output/17.png rename to seg_output/17.png diff --git a/f1_output/18.png b/seg_output/18.png similarity index 100% rename from f1_output/18.png rename to seg_output/18.png diff --git a/f1_output/19.png b/seg_output/19.png similarity index 100% rename from f1_output/19.png rename to seg_output/19.png diff --git a/f1_output/2.png b/seg_output/2.png similarity index 100% rename from f1_output/2.png rename to seg_output/2.png diff --git a/f1_output/20.png b/seg_output/20.png similarity index 100% rename from f1_output/20.png rename to seg_output/20.png diff --git a/f1_output/21.png b/seg_output/21.png similarity index 100% rename from f1_output/21.png rename to seg_output/21.png diff --git a/f1_output/22.png b/seg_output/22.png similarity index 100% rename from f1_output/22.png rename to seg_output/22.png diff --git a/f1_output/23.png b/seg_output/23.png similarity index 100% rename from f1_output/23.png rename to seg_output/23.png diff --git a/f1_output/24.png b/seg_output/24.png similarity index 100% rename from f1_output/24.png rename to seg_output/24.png diff --git a/f1_output/25.png b/seg_output/25.png similarity index 100% rename from f1_output/25.png rename to seg_output/25.png diff --git a/f1_output/26.png b/seg_output/26.png similarity index 100% rename from f1_output/26.png rename to seg_output/26.png diff --git a/f1_output/27.png b/seg_output/27.png similarity index 100% rename from f1_output/27.png rename to seg_output/27.png diff --git a/f1_output/28.png b/seg_output/28.png similarity index 100% rename from f1_output/28.png rename to seg_output/28.png diff --git a/f1_output/29.png b/seg_output/29.png similarity index 100% rename from f1_output/29.png rename to seg_output/29.png diff --git a/f1_output/3.png b/seg_output/3.png similarity index 100% rename from f1_output/3.png rename to seg_output/3.png diff --git a/f1_output/30.png b/seg_output/30.png similarity index 100% rename from f1_output/30.png rename to seg_output/30.png diff --git a/f1_output/4.png b/seg_output/4.png similarity index 100% rename from f1_output/4.png rename to seg_output/4.png diff --git a/f1_output/5.png b/seg_output/5.png similarity index 100% rename from f1_output/5.png rename to seg_output/5.png diff --git a/f1_output/6.png b/seg_output/6.png similarity index 100% rename from f1_output/6.png rename to seg_output/6.png diff --git a/f1_output/7.png b/seg_output/7.png similarity index 100% rename from f1_output/7.png rename to seg_output/7.png diff --git a/f1_output/8.png b/seg_output/8.png similarity index 100% rename from f1_output/8.png rename to seg_output/8.png diff --git a/f1_output/9.png b/seg_output/9.png similarity index 100% rename from f1_output/9.png rename to seg_output/9.png