Отслеживание и визуализация экспериментов YOLO с помощью Weights & Biases

Модели обнаружения объектов, такие как Ultralytics YOLO26, стали неотъемлемой частью многих приложений компьютерного зрения. Однако обучение, оценка и развертывание этих сложных моделей сопряжены с рядом проблем. Отслеживание ключевых метрик обучения, сравнение вариантов моделей, анализ поведения моделей и detect проблем требуют значительной инструментальной поддержки и управления экспериментами.

Смотреть: Как использовать Ultralytics YOLO26 с Weights and Biases

Это руководство демонстрирует интеграцию Ultralytics YOLO26 с Weights & Biases для улучшенного отслеживания экспериментов, сохранения контрольных точек модели и визуализации производительности модели. Оно также включает инструкции по настройке интеграции, обучению, тонкой настройке и визуализации результатов с использованием интерактивных функций Weights & Biases.

Weights & Biases

Weights & Biases — это передовая платформа MLOps, предназначенная для отслеживания, визуализации и управления экспериментами машинного обучения. Она включает автоматическое ведение журнала метрик обучения для полной воспроизводимости экспериментов, интерактивный пользовательский интерфейс для оптимизированного анализа данных и эффективные инструменты управления моделями для развертывания в различных средах.

Обучение YOLO26 с Weights & Biases

Вы можете использовать Weights & Biases для повышения эффективности и автоматизации процесса обучения YOLO26. Интеграция позволяет отслеживать эксперименты, сравнивать модели и принимать решения на основе данных для улучшения ваших проектов компьютерного зрения.

Установка

Чтобы установить необходимые пакеты, выполните команду:

# Install the required packages for Ultralytics YOLO and Weights & Biases
pip install -U ultralytics wandb

# Enable W&B logging for Ultralytics
yolo settings wandb=True

Для получения подробных инструкций и рекомендаций по процессу установки обязательно ознакомьтесь с нашим руководством по установке YOLO26. Если при установке необходимых пакетов для YOLO26 вы столкнетесь с какими-либо трудностями, обратитесь к нашему руководству по распространенным проблемам для поиска решений и советов.

Настройка Weights & Biases

После установки необходимых пакетов следующим шагом будет настройка вашей среды Weights & Biases. Это включает в себя создание учетной записи Weights & Biases и получение необходимого ключа API для обеспечения бесперебойной связи между вашей средой разработки и платформой W&B.

Начните с инициализации среды Weights & Biases в вашей рабочей области. Это можно сделать, выполнив следующую команду и следуя инструкциям.

import wandb

# Initialize your Weights & Biases environment
wandb.login(key="YOUR_API_KEY")

# Initialize your Weights & Biases environment
wandb login

Перейдите на страницу авторизации Weights & Biases, чтобы создать и получить свой API-ключ. Используйте этот ключ при появлении запроса на аутентификацию вашей среды с помощью W&B.

Использование: Обучение YOLO26 с Weights & Biases

Прежде чем приступать к инструкциям по использованию для обучения модели YOLO26 с Weights & Biases, обязательно ознакомьтесь с ассортиментом моделей YOLO26, предлагаемых Ultralytics. Это поможет вам выбрать наиболее подходящую модель для требований вашего проекта.

from ultralytics import YOLO

# Load a YOLO model
model = YOLO("yolo26n.pt")

# Train and Fine-Tune the Model
model.train(data="coco8.yaml", epochs=5, project="ultralytics", name="yolo26n")

# Train a YOLO26 model with Weights & Biases
yolo train data=coco8.yaml epochs=5 project=ultralytics name=yolo26n

Аргументы W&B

# Enable Weights & Biases logging
yolo settings wandb=True

# Disable Weights & Biases logging
yolo settings wandb=False

Понимание выходных данных

После запуска приведенного выше фрагмента кода, можно ожидать следующие ключевые результаты:

• Настройка нового запуска с его уникальным ID, указывающим на начало процесса обучения.
• Краткое описание структуры модели, включая количество слоев и параметров.
• Регулярные обновления важных метрик, таких как box loss, cls loss, dfl loss, precision, recall и показатели mAP, во время каждой эпохи обучения.
• В конце обучения отображаются подробные метрики, включая скорость инференса модели и общие метрики точности.
• Ссылки на панель управления Weights & Biases для углубленного анализа и визуализации процесса обучения, а также информация о местоположении локальных файлов журналов.

Просмотр панели управления Weights & Biases

После выполнения фрагмента кода использования вы можете получить доступ к панели мониторинга Weights & Biases (W&B) по предоставленной ссылке в выводе. Эта панель мониторинга предоставляет полный обзор процесса обучения вашей модели с YOLO26.

Ключевые особенности панели управления Weights & Biases

• Отслеживание метрик в реальном времени: Наблюдайте за метриками, такими как loss, accuracy и scores валидации, по мере их развития во время обучения, предлагая немедленные идеи для настройки модели. Узнайте, как отслеживаются эксперименты с помощью Weights & Biases.

• Оптимизация гиперпараметров: Weights & Biases помогает в тонкой настройке критически важных параметров, таких как скорость обучения, размер пакета и другие, повышая производительность YOLO26. Это помогает найти оптимальную конфигурацию для вашего конкретного набора данных и задачи.

• Сравнительный анализ: Платформа позволяет проводить параллельное сравнение различных прогонов обучения, что необходимо для оценки влияния различных конфигураций моделей и понимания того, какие изменения улучшают производительность.

• Визуализация прогресса обучения: Графическое представление ключевых метрик обеспечивает интуитивное понимание производительности модели по эпохам. Узнайте, как Weights & Biases помогает визуализировать результаты валидации.

• Мониторинг ресурсов: Следите за использованием CPU, GPU и памяти, чтобы оптимизировать эффективность процесса обучения и выявить потенциальные узкие места в вашем рабочем процессе.

• Управление артефактами модели: Получайте доступ к контрольным точкам модели и делитесь ими, что упрощает развертывание и сотрудничество с членами команды в сложных проектах.

• Просмотр результатов инференса с наложением изображений: Визуализируйте результаты прогнозирования на изображениях с помощью интерактивных наложений в Weights & Biases, обеспечивая четкое и подробное представление о производительности модели на реальных данных. Для получения более подробной информации см. возможности наложения изображений Weights & Biases.

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

Обзор

Это руководство помогло вам изучить интеграцию Ultralytics YOLO с Weights & Biases. Оно иллюстрирует способность этой интеграции эффективно отслеживать и визуализировать результаты обучения и прогнозирования моделей. Используя мощные функции W&B, вы можете оптимизировать свой рабочий процесс машинного обучения, принимать решения на основе данных и улучшать производительность своей модели.

Для получения более подробной информации об использовании посетите официальную документацию Weights & Biases или ознакомьтесь с презентацией Soumik Rakshit с YOLO VISION 2023 об этой интеграции.

Также обязательно ознакомьтесь со страницей руководства по интеграции Ultralytics, чтобы узнать больше о различных интересных интеграциях, таких как MLflow и Comet ML.

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

Как интегрировать Weights & Biases с Ultralytics YOLO26?

Для интеграции Weights & Biases с Ultralytics YOLO26:

1. Установите необходимые пакеты:

   pip install -U ultralytics wandb
   yolo settings wandb=True
   

2. Войдите в свою учетную запись Weights & Biases:

   import wandb
   
   wandb.login(key="YOUR_API_KEY")
   

3. Обучите вашу модель YOLO26 с включенным логированием W&B:

   from ultralytics import YOLO
   
   model = YOLO("yolo26n.pt")
   model.train(data="coco8.yaml", epochs=5, project="ultralytics", name="yolo26n")
   

Это автоматически зарегистрирует метрики, гиперпараметры и артефакты модели в вашем проекте W&B.

Каковы ключевые особенности интеграции Weights & Biases с YOLO26?

Ключевые особенности включают в себя:

• Отслеживание метрик в реальном времени во время обучения
• Инструменты оптимизации гиперпараметров
• Сравнительный анализ различных запусков обучения
• Визуализация прогресса обучения с помощью графиков
• Мониторинг ресурсов (CPU, GPU, использование памяти)
• Управление и совместное использование артефактов модели
• Просмотр результатов инференса с наложением изображений

Эти функции помогают отслеживать эксперименты, оптимизировать модели и более эффективно сотрудничать над проектами YOLO26.

Как просмотреть панель мониторинга Weights & Biases для обучения YOLO26?

После запуска скрипта обучения с интеграцией W&B:

1. Ссылка на вашу панель управления W&B будет предоставлена в выводе консоли.
2. Нажмите на ссылку или перейдите на wandb.ai и войдите в свою учетную запись.
3. Перейдите к своему проекту, чтобы просмотреть подробные метрики, визуализации и данные о производительности модели.

Панель мониторинга предоставляет информацию о процессе обучения вашей модели, позволяя эффективно анализировать и улучшать ваши модели YOLO26.

Можно ли отключить логирование Weights & Biases для обучения YOLO26?

Да, вы можете отключить логирование W&B, используя следующую команду:

yolo settings wandb=False

Чтобы снова включить логирование, используйте:

yolo settings wandb=True

Это позволяет вам контролировать, когда вы хотите использовать логирование W&B, не изменяя свои скрипты обучения.

Как Weights & Biases помогает в оптимизации моделей YOLO26?

Weights & Biases помогает оптимизировать модели YOLO26 путем:

1. Предоставления подробных визуализаций метрик обучения
2. Обеспечения простого сравнения различных версий моделей
3. Предлагая инструменты для настройки гиперпараметров
4. Обеспечение возможности совместного анализа производительности модели
5. Облегчение простого обмена артефактами и результатами модели

Эти функции помогают исследователям и разработчикам быстрее итерировать и принимать решения на основе данных для улучшения своих моделей YOLO26.

📅 Создано 2 лет назад ✏️ Обновлено 5 дней назад

Комментарии