allan87br commited on
Commit
d9bf098
·
verified ·
1 Parent(s): 0a2dcaf

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +108 -0
  2. requirements.txt +5 -0
app.py ADDED
@@ -0,0 +1,108 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import numpy as np
3
+ from scipy.ndimage import uniform_filter, median_filter, gaussian_filter1d
4
+ from skimage import exposure
5
+ import matplotlib.pyplot as plt
6
+ import io
7
+ from PIL import Image
8
+
9
+ def splitnormalize_balanced(img):
10
+ p_lo, p_hi = np.percentile(img, [2, 98])
11
+ img = np.clip((img - p_lo) / (p_hi - p_lo + 1e-6), 0, 1)
12
+ img = exposure.equalize_adapthist(img, clip_limit=0.003, nbins=256)
13
+ return img
14
+
15
+ def removebias(img, regionwidth=10, fraction=0.75):
16
+ kz = 2*regionwidth + 1
17
+ kx = 2*regionwidth + 1
18
+ bg = uniform_filter(img, size=(kz, kx))
19
+ out = img - fraction * bg
20
+ out = (out - out.min()) / (out.max() - out.min() + 1e-6)
21
+ return out
22
+
23
+ def estimate_medline_intensity(img):
24
+ H, W = img.shape
25
+ p1, p99 = np.percentile(img, [1, 99])
26
+ norm = np.clip((img - p1) / (p99 - p1 + 1e-6), 0, 1)
27
+ z_low = int(0.30 * H)
28
+ z_high = int(0.70 * H)
29
+ return np.argmax(norm[z_low:z_high, :], axis=0) + z_low
30
+
31
+ def detect_rpe_simple(img, medline):
32
+ H, W = img.shape
33
+ sm = gaussian_filter1d(img, sigma=4, axis=0)
34
+ grad = np.gradient(gaussian_filter1d(sm, sigma=1, axis=0), axis=0)
35
+ rpe = np.argmax(-grad[int(0.30*H):int(0.85*H), :], axis=0) + int(0.30*H)
36
+ return rpe
37
+
38
+ def linesweeter(y):
39
+ from scipy.signal import savgol_filter
40
+ return savgol_filter(y, 9, 2)
41
+
42
+ def overlay(img, curve, title=""):
43
+ H, W = img.shape
44
+ x = np.arange(W)
45
+ fig, ax = plt.subplots(figsize=(8,4))
46
+ ax.imshow(img, cmap="gray")
47
+ ax.plot(x, curve, 'r-', lw=2)
48
+ ax.set_title(title)
49
+ ax.axis('off')
50
+ buf = io.BytesIO()
51
+ plt.savefig(buf, format="png", bbox_inches="tight")
52
+ buf.seek(0)
53
+ return Image.open(buf)
54
+
55
+ with gr.Blocks(title="OCT Step-by-Step Visual Lab") as demo:
56
+ gr.Markdown("## 🧠 OCT Step-by-Step Visual Lab — Compare cada etapa lado a lado")
57
+
58
+ img_state = gr.State()
59
+
60
+ with gr.Tab("1) Carregar Imagem"):
61
+ img_input = gr.Image(label="Imagem OCT", type="numpy")
62
+ def store(img):
63
+ return img
64
+ img_input.change(store, inputs=img_input, outputs=img_state)
65
+
66
+ with gr.Tab("2) Normalização"):
67
+ btn_norm = gr.Button("Aplicar splitnormalize (balanced)")
68
+ out_norm = gr.Image(label="Resultado")
69
+ btn_norm.click(splitnormalize_balanced, inputs=img_state, outputs=out_norm)
70
+
71
+ with gr.Tab("3) Remove Bias"):
72
+ btn_bias = gr.Button("Aplicar removebias")
73
+ out_bias = gr.Image(label="Resultado")
74
+ btn_bias.click(removebias, inputs=img_state, outputs=out_bias)
75
+
76
+ with gr.Tab("4) Mediana 5×9"):
77
+ btn_med = gr.Button("Aplicar filtro mediano (5x9)")
78
+ out_med = gr.Image(label="Resultado")
79
+ btn_med.click(lambda i: median_filter(i, size=(5,9)), inputs=img_state, outputs=out_med)
80
+
81
+ with gr.Tab("5) Estimar Medline (IS/OS)"):
82
+ btn_medline = gr.Button("Calcular Medline")
83
+ out_medline = gr.Image(label="Visualização")
84
+ def show_medline(img):
85
+ med = estimate_medline_intensity(img)
86
+ return overlay(img, med, "Medline (IS/OS)")
87
+ btn_medline.click(show_medline, inputs=img_state, outputs=out_medline)
88
+
89
+ with gr.Tab("6) Detectar RPE"):
90
+ btn_rpe = gr.Button("Detectar RPE simples")
91
+ out_rpe = gr.Image(label="Visualização")
92
+ def step_rpe(img):
93
+ med = estimate_medline_intensity(img)
94
+ rpe = detect_rpe_simple(img, med)
95
+ return overlay(img, rpe, "RPE simples")
96
+ btn_rpe.click(step_rpe, inputs=img_state, outputs=out_rpe)
97
+
98
+ with gr.Tab("7) Suavização da RPE"):
99
+ btn_smooth = gr.Button("Suavizar RPE (linesweeter)")
100
+ out_smooth = gr.Image(label="Visualização")
101
+ def step_smooth(img):
102
+ med = estimate_medline_intensity(img)
103
+ rpe = detect_rpe_simple(img, med)
104
+ rpe_s = linesweeter(rpe)
105
+ return overlay(img, rpe_s, "RPE suavizada (linesweeter)")
106
+ btn_smooth.click(step_smooth, inputs=img_state, outputs=out_smooth)
107
+
108
+ demo.launch()
requirements.txt ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ gradio
2
+ numpy
3
+ scipy
4
+ scikit-image
5
+ matplotlib