Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -65,16 +65,16 @@ def segment(image):
|
|
| 65 |
mask_image = mask_image.astype(np.uint8)
|
| 66 |
mask_image = Image.fromarray(mask_image).convert("L")
|
| 67 |
|
| 68 |
-
#
|
| 69 |
positive_pixels = np.count_nonzero(mask_image)
|
| 70 |
total_pixels = mask_image.size[0] * mask_image.size[1]
|
| 71 |
percentage = (positive_pixels / total_pixels) * 100
|
| 72 |
|
| 73 |
-
|
| 74 |
-
|
| 75 |
-
|
| 76 |
|
| 77 |
-
return mask_image,
|
| 78 |
|
| 79 |
if __name__ == "__main__":
|
| 80 |
model = build_model(input_shape=(size, size, 1))
|
|
@@ -82,14 +82,14 @@ if __name__ == "__main__":
|
|
| 82 |
fn=segment,
|
| 83 |
inputs="image",
|
| 84 |
outputs=[
|
| 85 |
-
gr.Image(type="pil", label="
|
| 86 |
-
gr.Number(label="
|
| 87 |
-
gr.Number(label="
|
| 88 |
],
|
| 89 |
examples=[["benign (87).png"], ["benign (319).png"]],
|
| 90 |
-
title
|
| 91 |
-
description
|
| 92 |
-
|
| 93 |
-
|
| 94 |
"""
|
| 95 |
-
).launch(debug=True)
|
|
|
|
| 65 |
mask_image = mask_image.astype(np.uint8)
|
| 66 |
mask_image = Image.fromarray(mask_image).convert("L")
|
| 67 |
|
| 68 |
+
# Porcentagem de 0
|
| 69 |
positive_pixels = np.count_nonzero(mask_image)
|
| 70 |
total_pixels = mask_image.size[0] * mask_image.size[1]
|
| 71 |
percentage = (positive_pixels / total_pixels) * 100
|
| 72 |
|
| 73 |
+
# Calcular as porcentagens de 0 e 1
|
| 74 |
+
porcentagem_classe_0 = 100 - percentage
|
| 75 |
+
porcentagem_classe_1 = percentage
|
| 76 |
|
| 77 |
+
return mask_image, porcentagem_classe_0, porcentagem_classe_1
|
| 78 |
|
| 79 |
if __name__ == "__main__":
|
| 80 |
model = build_model(input_shape=(size, size, 1))
|
|
|
|
| 82 |
fn=segment,
|
| 83 |
inputs="image",
|
| 84 |
outputs=[
|
| 85 |
+
gr.Image(type="pil", label="Máscara de Câncer de Mama"),
|
| 86 |
+
gr.Number(label="Porcentagem da Classe 0"),
|
| 87 |
+
gr.Number(label="Porcentagem da Classe 1")
|
| 88 |
],
|
| 89 |
examples=[["benign (87).png"], ["benign (319).png"]],
|
| 90 |
+
title='<h1 style="text-align: center;">Segmentação de Imagens de Ultrassom de Câncer de Mama</h1>',
|
| 91 |
+
description="""
|
| 92 |
+
Foi desenvolvida uma demonstração de segmentação de imagens de ultrassom de câncer de mama.
|
| 93 |
+
Carregue um arquivo de imagem ou experimente um dos exemplos abaixo.
|
| 94 |
"""
|
| 95 |
+
).launch(debug=True)
|