Conjunto de dados de tumores cerebrais#

1,116

brain tumor

ultralytics

publicDetect

Brain Tumor is a compact MRI collection for localization-oriented medical imaging. Bounding-box labels distinguish positive and negative tumor targets, supporting experiments in lesion detection across varied brain anatomy, scan contrast, tumor appearance, and background tissue.

62 classes9.5 MBSep 2, 2026

negativepositive

O conjunto de dados de tumores cerebrais da Ultralytics é um conjunto de dados de deteção de objetos com 1.116 imagens médicas (893 para treino e 223 para validação) provenientes de exames de MRI e CT, anotadas em 2 classes: negative (sem tumor) e positive (com tumor). Permite treinar modelos de visão computacional para localizar tumores cerebrais em exames, apoiando o diagnóstico precoce e o planeamento do tratamento em aplicações de saúde.

Estrutura do Conjunto de Dados#

O conjunto de dados de tumores cerebrais contém 1.116 imagens divididas em dois subconjuntos predefinidos, definidos pela configuração brain-tumor.yaml:

Cada imagem é anotada com uma de 2 classes:

• negative: imagens sem tumor cerebral
• positive: imagens que mostram um tumor cerebral

O conjunto de dados é transferido automaticamente (4,21 MB) dos recursos do GitHub da Ultralytics na primeira vez que treinas, pelo que não é necessária qualquer configuração manual.

Explora Brain Tumor na Ultralytics Platform para percorreres as imagens com as respetivas sobreposições de anotações, veres a distribuição das classes e os mapas de calor das caixas delimitadoras no separador Gráficos, e fazeres uma cópia para treinares o teu próprio modelo na cloud.

Aplicações#

A deteção de tumores cerebrais com visão computacional permite o diagnóstico precoce, o planeamento do tratamento e a monitorização da progressão do tumor. Ao analisar exames de MRI ou CT, os modelos de deteção localizam os tumores com precisão, apoiando uma intervenção médica atempada e um tratamento personalizado.

Os profissionais de saúde podem tirar partido desta tecnologia para:

• Reduzir o tempo de diagnóstico e melhorar a precisão
• Apoiar o planeamento cirúrgico através da localização precisa dos tumores
• Monitorizar a eficácia do tratamento ao longo do tempo
• Apoiar a investigação em oncologia e neurologia

YAML do Conjunto de Dados#

Um ficheiro YAML define a configuração do conjunto de dados, incluindo caminhos, classes e outras informações relevantes. Para o conjunto de dados de tumores cerebrais, o ficheiro brain-tumor.yaml é mantido em https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/brain-tumor.yaml.

# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/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.21 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: brain-tumor # dataset root dir
train: images/train # train images (relative to 'path') 893 images
val: images/val # val images (relative to 'path') 223 images

# Classes
names:
  0: negative
  1: positive

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

Utilização#

Para treinares um modelo YOLO26 no conjunto de dados de tumores cerebrais durante 100 épocas, com um tamanho de imagem de 640, utiliza os fragmentos de código fornecidos. Para obteres uma lista detalhada dos argumentos disponíveis, consulta a página de Treino do modelo.

from ultralytics import YOLO

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

# Train the model
results = model.train(data="brain-tumor.yaml", 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")

Imagens de exemplo e anotações#

O conjunto de dados de tumores cerebrais contém exames cerebrais de MRI e CT com e sem tumores. Abaixo, é apresentada uma imagem de exemplo do conjunto de dados com as respetivas anotações.

• Imagem em mosaico: este lote de treino mostra imagens do conjunto de dados dispostas em mosaico. A criação de mosaicos combina várias imagens numa só durante o treino, aumentando a diversidade do lote para que o modelo generalize melhor entre diferentes tamanhos, formas e localizações de tumores na análise de imagens médicas.

Citações e agradecimentos#

O conjunto de dados foi disponibilizado ao abrigo da Licença AGPL-3.0.

Se utilizares este conjunto de dados no teu trabalho de investigação ou desenvolvimento, cita-o adequadamente:

@dataset{Ultralytics_Brain_Tumor_Dataset_2023,
    author = {Ultralytics},
    title = {Brain Tumor Detection Dataset},
    year = {2023},
    publisher = {Ultralytics},
    url = {https://docs.ultralytics.com/datasets/detect/brain-tumor}
}

Perguntas frequentes#