vit-xray-pneumonia-classification

lxyuan

Clasificación de imagen

Este modelo es una versión ajustada de google/vit-base-patch16-224-in21k en el conjunto de datos chest-xray-classification. Logra los siguientes resultados en el conjunto de evaluación: Pérdida: 0.0868 Precisión: 0.9742 Este modelo puede ser utilizado para la clasificación de imágenes de radiografías de tórax, diferenciando entre neumonía y normal.

Como usar

from transformers import pipeline
classifier = pipeline(model="lxyuan/vit-xray-pneumonia-classification")

# Imagen tomada de https://www.news-medical.net/health/What-is-Viral-Pneumonia.aspx
classifier("https://d2jx2rerrg6sh3.cloudfront.net/image-handler/ts/20200618040600/ri/650/picture/2020/6/shutterstock_786937069.jpg")

>>>
[{'score': 0.990334689617157, 'label': 'PNEUMONIA'},
{'score': 0.009665317833423615, 'label': 'NORMAL'}]

Funcionalidades

Clasificación de imágenes de radiografías de tórax
Utiliza Transformers
Desarrollado con PyTorch
Soporte para TensorBoard y Safetensors
Entrenado con el conjunto de datos chest-xray-classification

Casos de uso

Diagnóstico de neumonía a partir de radiografías de tórax
Clasificación automática de radiografías en centros de salud
Asistencia a radiólogos en la detección de neumonía