Christhofer commited on
Commit
2c78adf
·
1 Parent(s): 7996686

removing unused code in app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -32
app.py CHANGED
@@ -70,12 +70,6 @@ def cargar_dicom_file(image_bytes, tipo):
70
  dcm = pydicom.dcmread(image_bytes)
71
  numpy_lossy_image = dcm.pixel_array
72
  return numpy_lossy_image
73
- #if tipo != 'storage':
74
- # lossy_image = tfio.image.decode_dicom_image(image_bytes.getvalue(), scale='auto', on_error='lossy', dtype=tf.uint8)
75
- #else:
76
- # lossy_image = tfio.image.decode_dicom_image(image_bytes, scale='auto', on_error='lossy', dtype=tf.uint8)
77
- #numpy_lossy_image = np.squeeze(lossy_image.numpy())
78
- #return numpy_lossy_image
79
 
80
  def mostrar_resultados_prediccion(crop_image):
81
  resultado_alexnet = modelo_alexnet.predecir(crop_image)
@@ -97,24 +91,6 @@ def obtener_informacion_archivo_dicom(image_bytes, tipo):
97
  }
98
  return data
99
 
100
- def guardar_correccion():
101
- df_resultados = dict()
102
-
103
- df_resultados['resultado_alexnet_benigno'] = resultado_alexnet[0]
104
- df_resultados['resultado_alexnet_maligno'] = resultado_alexnet[1]
105
- df_resultados['resultado_alexnet_normal'] = resultado_alexnet[2]
106
- df_resultados['resultado_resnet50_benigno'] = resultado_resnet50[0]
107
- df_resultados['resultado_resnet50_maligno'] = resultado_resnet50[1]
108
- df_resultados['resultado_resnet50_normal'] = resultado_resnet50[2]
109
- df_resultados['correccion_benigno'] = st.session_state['correcion_benigno']
110
- df_resultados['correccion_maligno'] = st.session_state['correcion_maligno']
111
- df_resultados['correccion_normal'] = st.session_state['correcion_normal']
112
- #df = pd.read_csv('./resultados_pruebas.csv')
113
- df_resultados = pd.DataFrame(df_resultados, index=[0])
114
- #df_completo = pd.concat([df, df_resultados], axis=0)
115
- print(df_resultados)
116
- df_resultados.to_csv('./resultados_pruebas.csv')
117
-
118
 
119
  st.write("# Modelo de Machine Learning para el diagnostico del cancer de mama")
120
  uploaded_file = st.file_uploader("Subir archivo DICOM", type=[".dcm"])
@@ -146,11 +122,4 @@ if uploaded_file:
146
  st.write("### Probabilidades según tipo de tumor - AlexNet")
147
  st.bar_chart(df_alexnet)
148
  st.write("### Probabilidades según tipo de tumor - ResNet50")
149
- st.bar_chart(df_resnet50)
150
- df_form = obtener_informacion_archivo_dicom(uploaded_file, tipo='')
151
- with st.form("Correcion") as form:
152
- st.write("#### Tipo de tumorx")
153
- st.slider("Benigno", 0, 100, 0, key='correcion_benigno')
154
- st.slider("Maligno", 0, 100, 0, key='correcion_maligno')
155
- st.slider("Normal", 0, 100, 0, key='correcion_normal')
156
- submitted = st.form_submit_button("Realizar correccion", on_click=guardar_correccion)
 
70
  dcm = pydicom.dcmread(image_bytes)
71
  numpy_lossy_image = dcm.pixel_array
72
  return numpy_lossy_image
 
 
 
 
 
 
73
 
74
  def mostrar_resultados_prediccion(crop_image):
75
  resultado_alexnet = modelo_alexnet.predecir(crop_image)
 
91
  }
92
  return data
93
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
94
 
95
  st.write("# Modelo de Machine Learning para el diagnostico del cancer de mama")
96
  uploaded_file = st.file_uploader("Subir archivo DICOM", type=[".dcm"])
 
122
  st.write("### Probabilidades según tipo de tumor - AlexNet")
123
  st.bar_chart(df_alexnet)
124
  st.write("### Probabilidades según tipo de tumor - ResNet50")
125
+ st.bar_chart(df_resnet50)