Vai al contenuto

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.



Guarda: Rilevamento del tumore al cervello con Ultralytics HUB

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/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

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.

Esempio di treno

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

Esempio di inferenza

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 campione del set di dati sui tumori cerebrali

  • 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 .

DOMANDE FREQUENTI

Qual è la struttura del dataset dei tumori cerebrali disponibile nella documentazione di Ultralytics ?

Il set di dati sui tumori cerebrali è diviso in due sottoinsiemi: il set di addestramento è composto da 893 immagini con annotazioni corrispondenti, mentre il set di test comprende 223 immagini con annotazioni accoppiate. Questa divisione strutturata aiuta a sviluppare modelli di visione artificiale robusti e accurati per il rilevamento dei tumori cerebrali. Per maggiori informazioni sulla struttura del set di dati, visita la sezione Struttura del set di dati.

Come posso addestrare un modello YOLOv8 sul dataset dei tumori cerebrali utilizzando Ultralytics?

Puoi addestrare un modello YOLOv8 sul dataset dei tumori cerebrali per 100 epoch con un'immagine di 640px utilizzando sia il metodo Python che CLI . Di seguito sono riportati gli esempi per entrambi:

Esempio di treno

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

Per un elenco dettagliato degli argomenti disponibili, consulta la pagina Formazione.

Quali sono i vantaggi dell'utilizzo del set di dati sui tumori cerebrali per l'IA nell'assistenza sanitaria?

L'utilizzo del set di dati sui tumori cerebrali nei progetti di intelligenza artificiale consente di effettuare diagnosi precoci e pianificare il trattamento dei tumori cerebrali. Aiuta ad automatizzare l'identificazione dei tumori cerebrali attraverso la computer vision, facilitando interventi medici accurati e tempestivi e supportando strategie di trattamento personalizzate. Questa applicazione ha un potenziale significativo per migliorare i risultati dei pazienti e l'efficienza medica.

Come si esegue l'inferenza utilizzando un modello YOLOv8 ottimizzato sul set di dati dei tumori cerebrali?

L'inferenza con un modello YOLOv8 ottimizzato può essere eseguita sia con l'approccio Python che con quello CLI . Ecco gli esempi:

Esempio di inferenza

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"

Dove posso trovare la configurazione YAML per il dataset dei tumori cerebrali?

Il file di configurazione YAML per il dataset del tumore al cervello si trova all'indirizzo brain-tumor.yaml. Questo file include i percorsi, le classi e altre informazioni rilevanti necessarie per l'addestramento e la valutazione dei modelli su questo set di dati.



Creato 2024-03-19, Aggiornato 2024-07-04
Autori: glenn-jocher (3), RizwanMunawar (1), Burhan-Q (1)

Commenti