Conjunto de datos sobre tumores cerebrales

Un conjunto de datos de detección de tumores cerebrales consiste en imágenes médicas procedentes de resonancias magnéticas o tomografías computarizadas, que contienen información sobre la presencia, la ubicación y las características de los tumores cerebrales. Este conjunto de datos es esencial para entrenar algoritmos de visión por ordenador que automaticen la identificación de tumores cerebrales y contribuyan a un diagnóstico precoz y a la planificación del tratamiento.

Observa: Detección de tumores cerebrales mediante Ultralytics HUB

Estructura del conjunto de datos

El conjunto de datos sobre tumores cerebrales se divide en dos subconjuntos:

• Conjunto de entrenamiento: Compuesto por 893 imágenes, cada una acompañada de sus correspondientes anotaciones.
• Conjunto de pruebas: Compuesto por 223 imágenes, con anotaciones emparejadas para cada una de ellas.

Aplicaciones

La detección de tumores cerebrales mediante visión por ordenador permite el diagnóstico precoz, la planificación del tratamiento y el seguimiento de la progresión del tumor. Mediante el análisis de datos de imágenes médicas como resonancias magnéticas o tomografías computarizadas, los sistemas de visión por ordenador ayudan a identificar con precisión los tumores cerebrales, lo que contribuye a una intervención médica oportuna y a estrategias de tratamiento personalizadas.

Conjunto de datos YAML

Para definir la configuración del conjunto de datos se utiliza un archivo YAML (Yet Another Markup Language). Contiene información sobre las rutas del conjunto de datos, las clases y otra información relevante. En el caso del conjunto de datos sobre tumores cerebrales, el archivo brain-tumor.yaml se mantiene en https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/brain-tumor.yaml.

# Ultralytics YOLO 🚀, AGPL-3.0 license
# Brain-tumor dataset by Ultralytics
# Documentation: https://docs.ultralytics.com/datasets/detect/brain-tumor/
# Example usage: yolo train data=brain-tumor.yaml
# parent
# ├── ultralytics
# └── datasets
#     └── brain-tumor  ← downloads here (4.05 MB)

# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
path: ../datasets/brain-tumor # dataset root dir
train: train/images # train images (relative to 'path') 893 images
val: valid/images # val images (relative to 'path') 223 images
test: # test images (relative to 'path')

# Classes
names:
  0: negative
  1: positive

# Download script/URL (optional)
download: https://github.com/ultralytics/assets/releases/download/v0.0.0/brain-tumor.zip

Utilización

Para entrenar un modelo YOLO11n en el conjunto de datos de tumores cerebrales durante 100 épocas con un tamaño de imagen de 640, utilice los fragmentos de código proporcionados. Para obtener una lista detallada de los argumentos disponibles, consulte la página de entrenamiento del modelo.

from ultralytics import YOLO

# Load a model
model = YOLO("yolo11n.pt")  # load a pretrained model (recommended for training)

# Train the model
results = model.train(data="brain-tumor.yaml", epochs=100, imgsz=640)

# Start training from a pretrained *.pt model
yolo detect train data=brain-tumor.yaml model=yolo11n.pt epochs=100 imgsz=640

from ultralytics import YOLO

# Load a model
model = YOLO("path/to/best.pt")  # load a brain-tumor fine-tuned model

# Inference using the model
results = model.predict("https://ultralytics.com/assets/brain-tumor-sample.jpg")

# Start prediction with a finetuned *.pt model
yolo detect predict model='path/to/best.pt' imgsz=640 source="https://ultralytics.com/assets/brain-tumor-sample.jpg"

Ejemplos de imágenes y anotaciones

El conjunto de datos sobre tumores cerebrales abarca una amplia gama de imágenes con diversas categorías de objetos y escenas complejas. A continuación se presentan ejemplos de imágenes del conjunto de datos, acompañadas de sus respectivas anotaciones

• Imagen en mosaico: Aquí se muestra un lote de entrenamiento compuesto por imágenes del conjunto de datos en mosaico. El mosaico, una técnica de entrenamiento, consolida múltiples imágenes en una, mejorando la diversidad del lote. Este enfoque ayuda a mejorar la capacidad del modelo para generalizar a través de varios tamaños de objetos, relaciones de aspecto y contextos.

Este ejemplo pone de relieve la diversidad y complejidad de las imágenes del conjunto de datos sobre tumores cerebrales, y subraya las ventajas de incorporar mosaicos durante la fase de formación.

Citas y agradecimientos

El conjunto de datos se ha publicado bajo la licenciaAGPL-3.0 .

PREGUNTAS FRECUENTES

¿Cuál es la estructura del conjunto de datos sobre tumores cerebrales disponible en la documentación de Ultralytics ?

El conjunto de datos sobre tumores cerebrales se divide en dos subconjuntos: el conjunto de entrenamiento consta de 893 imágenes con sus correspondientes anotaciones, mientras que el conjunto de pruebas comprende 223 imágenes con anotaciones emparejadas. Esta división estructurada ayuda a desarrollar modelos de visión por ordenador robustos y precisos para detectar tumores cerebrales. Para más información sobre la estructura del conjunto de datos, visite la sección Estructura del conjunto de datos.

¿Cómo puedo entrenar un modelo YOLO11 en el conjunto de datos de tumores cerebrales utilizando Ultralytics?

Puede entrenar un modelo YOLO11 en el conjunto de datos de tumores cerebrales durante 100 épocas con un tamaño de imagen de 640px utilizando los métodos Python y CLI . A continuación se muestran los ejemplos de ambos:

from ultralytics import YOLO

# Load a model
model = YOLO("yolo11n.pt")  # load a pretrained model (recommended for training)

# Train the model
results = model.train(data="brain-tumor.yaml", epochs=100, imgsz=640)

# Start training from a pretrained *.pt model
yolo detect train data=brain-tumor.yaml model=yolo11n.pt epochs=100 imgsz=640

Para obtener una lista detallada de los argumentos disponibles, consulte la página Formación.

¿Cuáles son las ventajas de utilizar el conjunto de datos sobre tumores cerebrales para la IA en la atención sanitaria?

El uso del conjunto de datos sobre tumores cerebrales en proyectos de IA permite el diagnóstico precoz y la planificación del tratamiento de tumores cerebrales. Ayuda a automatizar la identificación de tumores cerebrales mediante visión por ordenador, facilitando intervenciones médicas precisas y oportunas y apoyando estrategias de tratamiento personalizadas. Esta aplicación tiene un gran potencial para mejorar los resultados de los pacientes y la eficiencia médica.

¿Cómo se realiza la inferencia utilizando un modelo YOLO11 afinado en el conjunto de datos de tumores cerebrales?

La inferencia mediante un modelo YOLO11 ajustado puede realizarse con los enfoques Python o CLI . He aquí los ejemplos:

from ultralytics import YOLO

# Load a model
model = YOLO("path/to/best.pt")  # load a brain-tumor fine-tuned model

# Inference using the model
results = model.predict("https://ultralytics.com/assets/brain-tumor-sample.jpg")

# Start prediction with a finetuned *.pt model
yolo detect predict model='path/to/best.pt' imgsz=640 source="https://ultralytics.com/assets/brain-tumor-sample.jpg"

¿Dónde puedo encontrar la configuración YAML para el conjunto de datos de tumores cerebrales?

El archivo de configuración YAML para el conjunto de datos de tumores cerebrales se encuentra en brain-tumor.yaml. Este archivo incluye rutas, clases e información adicional necesaria para entrenar y evaluar modelos en este conjunto de datos.

Creado hace 8 meses ✏️ Actualizado hace 2 meses

Comentarios