Set di dati sul tumore al cervello

Un set di dati per il rilevamento dei tumori cerebrali è costituito da immagini mediche provenienti da risonanza magnetica o TAC, contenenti informazioni sulla presenza, la posizione e le caratteristiche del tumore cerebrale. Questo set di dati è essenziale per l'addestramento di algoritmi di computer vision per automatizzare l'identificazione dei tumori cerebrali, favorendo la diagnosi precoce e la pianificazione del trattamento.

Struttura del set di dati

Il set di dati sui tumori cerebrali è suddiviso in due sottoinsiemi:

• Set di allenamento: Consiste in 893 immagini, ognuna accompagnata da annotazioni corrispondenti.
• Set di prova: Composto da 223 immagini, con annotazioni abbinate per ciascuna di esse.

Applicazioni

L'applicazione del rilevamento dei tumori cerebrali tramite la computer vision consente la diagnosi precoce, la pianificazione del trattamento e il monitoraggio della progressione del tumore. Analizzando i dati di imaging medico come la risonanza magnetica o la TAC, i sistemi di computer vision aiutano a identificare con precisione i tumori cerebrali, favorendo un intervento medico tempestivo e strategie di trattamento personalizzate.

Set di dati YAML

Un file YAML (Yet Another Markup Language) viene utilizzato per definire la configurazione del dataset. Contiene informazioni sui percorsi, le classi e altre informazioni rilevanti del dataset. Nel caso del set di dati sui tumori cerebrali, il file brain-tumor.yaml Il file viene mantenuto all'indirizzo 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://ultralytics.com/assets/brain-tumor.zip

Utilizzo

Per addestrare un modello YOLOv8n sul dataset dei tumori cerebrali per 100 epoch con una dimensione dell'immagine di 640, utilizza i frammenti di codice forniti. Per un elenco dettagliato degli argomenti disponibili, consulta la pagina di addestramento del modello.

from ultralytics import YOLO

# Load a model
model = YOLO('yolov8n.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=yolov8n.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"

Immagini di esempio e annotazioni

Il set di dati sui tumori cerebrali comprende un'ampia gamma di immagini con diverse categorie di oggetti e scene complesse. Di seguito sono riportati alcuni esempi di immagini tratte dal set di dati, accompagnate dalle rispettive annotazioni

• Immagine mosaicata: Qui viene mostrato un lotto di formazione che comprende immagini del set di dati mosaicate. La mosaicatura, una tecnica di formazione, consolida più immagini in una sola, migliorando la diversità del lotto. Questo approccio aiuta a migliorare la capacità del modello di generalizzare tra oggetti di varie dimensioni, rapporti di aspetto e contesti.

Questo esempio evidenzia la diversità e l'intricatezza delle immagini all'interno del set di dati sul tumore al cervello, sottolineando i vantaggi dell'incorporazione della mosaicatura durante la fase di addestramento.

Citazioni e ringraziamenti

Il set di dati è stato rilasciato con la licenzaAGPL-3.0 .

Commenti