Датасет опухолей головного мозга#

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

Датасет опухолей головного мозга Ultralytics — это датасет для обнаружения объектов, содержащий 1 116 медицинских изображений (893 для обучения и 223 для валидации), полученных с помощью МРТ- и КТ-сканирования и размеченных по 2 классам: negative (опухоли нет) и positive (опухоль присутствует). Он позволяет обучать модели компьютерного зрения для обнаружения опухолей головного мозга на снимках, поддерживая раннюю диагностику и планирование лечения в медицинских приложениях.

Структура датасета#

Датасет опухолей головного мозга содержит 1 116 изображений, разделённых на два заранее определённых поднабора согласно конфигурации brain-tumor.yaml:

Каждое изображение размечено одним из 2 классов:

• negative: изображения без опухоли головного мозга
• positive: изображения с опухолью головного мозга

При первом запуске обучения датасет автоматически скачивается (4,21 МБ) из ресурсов Ultralytics на GitHub, поэтому ручная настройка не требуется.

Открой датасет опухолей головного мозга на платформе Ultralytics, чтобы просмотреть изображения с наложенными аннотациями, изучить распределение классов и тепловые карты ограничивающих рамок на вкладке Charts, а также клонировать датасет для обучения собственной модели в облаке.

Применения#

Обнаружение опухолей головного мозга с помощью компьютерного зрения обеспечивает раннюю диагностику, планирование лечения и мониторинг прогрессирования опухоли. Анализируя МРТ- или КТ-снимки, модели обнаружения точно определяют расположение опухолей, поддерживая своевременное медицинское вмешательство и персонализированное лечение.

Медицинские специалисты могут использовать эту технологию, чтобы:

• Сократить время диагностики и повысить точность
• Помочь в планировании хирургических операций за счёт точного определения расположения опухолей
• Отслеживать эффективность лечения с течением времени
• Поддерживать исследования в области онкологии и неврологии

YAML датасета#

Файл YAML определяет конфигурацию датасета, включая пути, классы и другую актуальную информацию. Для датасета опухолей головного мозга файл brain-tumor.yaml хранится по адресу 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

Использование#

Чтобы обучить модель YOLO26 на датасете опухолей головного мозга в течение 100 эпох с размером изображения 640, используй приведённые фрагменты кода. Подробный список доступных аргументов смотри на странице обучения модели.

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")

Примеры изображений и аннотаций#

Датасет опухолей головного мозга содержит МРТ- и КТ-снимки головного мозга с опухолями и без них. Ниже приведён пример изображения из датасета с его аннотациями.

• Изображение с мозаикой: этот пакет данных для обучения содержит изображения датасета, объединённые в мозаику. Мозаика объединяет несколько изображений в одно во время обучения, повышая разнообразие пакета, чтобы модель лучше обобщала данные при разных размерах, формах и расположении опухолей для анализа медицинских изображений.

Цитирование и благодарности#

Датасет доступен по лицензии AGPL-3.0.

Если ты используешь этот датасет в исследовательской или разработческой работе, пожалуйста, корректно укажи его в цитировании:

@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}
}

Часто задаваемые вопросы#