Detector de COVID-19 usando imágenes de Rayos X

Marzo 28, 2020

Si te dijera que tu puedes hacer tu propio modelo de vision computacional, sin tener conocimientos previos en programación, ¿ me creerías ?

Quiero explicarte como puedes aplicar un algoritmo de clasificación para cualquier tema de tu interés, en este caso nos orientaremos a las pruebas realizadas de casos positivos del covid-19, teniendo como dataset imágenes de algunas radiografías.

El dataset con el cual estaremos trabajando es tomado del siguiente repositorio dataset covid-19, esta base de datos se encuentra en construcción y día tras día le estarán agregando mas datos.

Una vez descargado el repositorio nos dirigimos a la carpeta imágenes, guardamos todas las imágenes dominadas covid-19, esta es una de las radiografías con respecto al virus.

radiografías  del coronavirus

Teachable machine es una herramienta que nos permite realizar nuestros propios algoritmos de clasificación, ( si quieres saber mas sobre esta herramienta visita el siguiente enlace Teachable machine)

Su interfaz luce de la siguiente manera.

Teachablemachine coronavirus

Como se puede observar podemos crear 3 clases de proyectos con respecto algoritmos de clasificación, imágenes,audio,actitud.

Nuestro proyecto estará enfocado a imágenes, por defecto encontraremos 2 clases de grupos, el numero de grupos sera establecido según la necesidad del proyecto, en nuestro caso utilizaremos solamente 2.

Teachablemachine coronavirus

Cuando trabajamos con algoritmos de predicción, debemos tener en cuenta que necesitamos saber con respecto a nuestra base de datos, como se van a distribuir nuestras imágenes, en nuestra variable train debemos tener 2 carpetas, una denominada Covid y otra noCovid.

En nuestra carpeta covid guardamos todas las imágenes que tengan la etiqueta covid-19, en nuestra carpeta noCovid tendremos imágenes de otras radiografías sin covid-19, es importante aclarar que por el momento solo contamos con 21 imágenes con casos positivos.

como vamos a necesitar una imagen de validación para que el algoritmo nos clasifique su clase, creamos otra carpeta que se llame validation, donde le pasaremos una sola imagen de un caso positivo, esta imagen no debe estar en nuestra carpeta covid, ya que es con la que vamos a validar si el algoritmo esta clasificando bien, si no la sacamos de la carpeta covid nos va arrogar una probabilidad de 100% de caso positivo.

imágenes de covid-19

Ahora podemos empezar a importar nuestras imágenes a nuestras 2 clases.

Teachable machine coronavirus

Al modelo le pasamos nuestra imagen de valitation para que la clasifique según los datos de entrenamiento.

Teachable machine coronavirus

Nos da como resultado el 97 % como caso positivo de coronavirus, debemos tener presente que no contamos con una base de datos de miles de imágenes, esto quiere decir que el algoritmo no es de total confianza, ya que solo lo alimentamos con 20 imágenes con casos positivos, esto es una pequeña prueba de lo que se puede llegar hacer con algoritmos de clasificación.

El siguiente paso es que tu hagas tu propio proyecto, o puedes buscar mas base de datos de radiografías y mejorar esta prueba.

@Sergiorubiano25