Clasificación de Imágenes de Radiografías de Torax de Pacientes con Covid, con Neumonía y sin padecimientos

Usando el dataset en Kaggle de imágenes de tórax para pacientes se hizo modelo basado en el modelo de Google mobilenet_v2 para categorizar si es que se trataba de una imagen con Covid19, con neumonía típica o si no tenía ninguno de estos síntomas

Ya que en el dataset no se incluían muchas imágenes se decidió hacer un aumento de datos haciendo diferentes distorsiones en las imágenes como rotación, zoom, movimiento, etc. así estas imágenes distorsionadas las podemos tomar como nuevos datos.

Durante el entrenamiento de el modelo vimos que al aumentar el número de épocas sí aumentaba la precisión de la predicción con los datos de entrenamiento pero iba disminuyendo con los datos de prueba por lo que estábamos teniendo un sobre-ajuste, es por esto que decidimos dejar el número de épocas de entrenamiento con un valor más bajo, haciendo este cambio pasamos de una precisión de 85% a 93%.

Haciendo las pruebas obtenemos un buen resultado