Commit
·
2c78adf
1
Parent(s):
7996686
removing unused code in app.py
Browse files
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)
|
|
|
|
|
|
|
|
|
|
|
|
|
|