DHEIVER commited on
Commit
32113de
·
1 Parent(s): 6b15a22

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -13
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
- #Porcentaje 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 los porcentajes de 0 y 1
74
- class_0_percentage = 100 - percentage
75
- class_1_percentage = percentage
76
 
77
- return mask_image, class_0_percentage, class_1_percentage
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="Breast Cancer Mask"),
86
- gr.Number(label="Class 0 Percentage"),
87
- gr.Number(label="Class 1 Percentage")
88
  ],
89
  examples=[["benign (87).png"], ["benign (319).png"]],
90
- title = '<h1 style="text-align: center;">Breast Cancer Ultrasound Image Segmentation </h1>',
91
- description = """
92
- Se ha desarrollado una demostración de segmentación de imágenes de ultrasonido de cáncer de mama.
93
- Cargue un archivo de imagen o pruebe uno de los ejemplos a continuación
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)