Тренировка Elevating YOLOv8 : Упрости свой процесс записи логов с помощью Comet ML

Регистрация ключевых деталей обучения, таких как параметры, метрики, предсказания изображений и контрольные точки модели, очень важна в машинном обучении - это позволяет сделать проект прозрачным, прогресс измеримым, а результаты - повторяемыми.

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

Comet ML

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

Используй мощь YOLOv8 и Comet ML

Объединив Ultralytics YOLOv8 с Comet ML, ты откроешь целый ряд преимуществ. К ним относятся упрощенное управление экспериментами, понимание в реальном времени для быстрой корректировки, гибкие и индивидуальные опции ведения журнала, а также возможность вести журнал экспериментов в автономном режиме, когда доступ к интернету ограничен. Эта интеграция позволяет тебе принимать решения на основе данных, анализировать показатели эффективности и добиваться исключительных результатов.

Установка

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

# Install the required packages for YOLOv8 and Comet ML
pip install ultralytics comet_ml torch torchvision

Настройка Comet ML

После установки необходимых пакетов тебе нужно будет зарегистрироваться, получить API-ключComet и настроить его.

# Set your Comet Api Key
export COMET_API_KEY=<Your API Key>

Затем ты можешь инициализировать свой проект Comet. Comet автоматически определит ключ API и приступит к настройке.

import comet_ml

comet_ml.login(project_name="comet-example-yolov8-coco128")

Если ты используешь ноутбук Google Colab, то приведенный выше код попросит тебя ввести API-ключ для инициализации.

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

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

from ultralytics import YOLO

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

# Train the model
results = model.train(
    data="coco8.yaml",
    project="comet-example-yolov8-coco128",
    batch=32,
    save_period=1,
    save_json=True,
    epochs=3,
)

После запуска обучающего кода Comet ML создаст эксперимент в твоем рабочем пространстве Comet , чтобы автоматически отслеживать ход выполнения. Затем тебе будет предоставлена ссылка, по которой ты сможешь просмотреть подробный лог процесса обучения твоей YOLOv8 модели.

Comet автоматически регистрирует следующие данные без дополнительной настройки: метрики, такие как mAP и потери, гиперпараметры, контрольные точки модели, интерактивную матрицу путаницы и предсказания границ изображения.

Пойми эффективность своей модели с помощью Comet ML-визуализации

Давайте углубимся в то, что вы увидите на Comet Панель управления машинным обучением после того, как вы YOLOv8 Модель начинает обучение. Панель мониторинга — это место, где происходят все действия, представляя ряд автоматически регистрируемой информации с помощью визуальных элементов и статистики. Вот краткий обзор:

Панели для экспериментов

В разделе "Панели экспериментов" панели управления Comet ML собраны и представлены различные прогоны и их метрики, такие как потеря маски сегмента, потеря класса, точность и средняя точность.

Метрика

В разделе метрик у тебя есть возможность изучить метрики и в табличном формате, который отображается в специальной панели, как показано здесь.

Интерактивная матрица запутывания

Матрица путаницы, расположенная на вкладке Confusion Matrix, представляет собой интерактивный способ оценить точность классификации модели. Она подробно описывает правильные и неправильные предсказания, позволяя тебе понять сильные и слабые стороны модели.

Системные показатели

Comet ML регистрирует системные показатели, чтобы помочь выявить узкие места в процессе обучения. Он включает в себя такие метрики, как использование GPU , использование памяти GPU , использование CPU и использование оперативной памяти. Они необходимы для контроля эффективности использования ресурсов во время обучения модели.

Настройка журнала Comet ML

Comet ML предлагает гибко настраивать поведение журнала, задавая переменные окружения. Эти настройки позволяют тебе адаптировать Comet ML к твоим конкретным потребностям и предпочтениям. Вот несколько полезных вариантов настройки:

Предсказания образа логгинга

Ты можешь контролировать количество прогнозов изображений, которые Comet ML регистрирует во время экспериментов. По умолчанию Comet ML регистрирует 100 предсказаний изображений из валидационного набора. Однако ты можешь изменить это число, чтобы оно лучше соответствовало твоим требованиям. Например, чтобы записать в журнал 200 предсказаний изображений, используй следующий код:

import os

os.environ["COMET_MAX_IMAGE_PREDICTIONS"] = "200"

Интервал пакетного протоколирования

Comet ML позволяет тебе указать, как часто будут регистрироваться партии предсказаний изображений. На сайте COMET_EVAL_BATCH_LOGGING_INTERVAL переменная окружения управляет этой частотой. По умолчанию установлено значение 1, при котором в журнал записываются предсказания из каждой партии валидации. Ты можешь изменить это значение, чтобы регистрировать предсказания с другим интервалом. Например, если установить значение 4, то в журнал будут записываться предсказания из каждой четвертой партии.

import os

os.environ["COMET_EVAL_BATCH_LOGGING_INTERVAL"] = "4"

Отключение ведения журнала матрицы запутывания

В некоторых случаях ты можешь не выводить матрицу путаницы из твоего валидационного набора после каждой эпохи. Ты можешь отключить эту функцию, установив параметр COMET_EVAL_LOG_CONFUSION_MATRIX переменную окружения на "false". Матрица путаницы будет записана в журнал только один раз, после завершения тренировки.

import os

os.environ["COMET_EVAL_LOG_CONFUSION_MATRIX"] = "false"

Ведение журнала в автономном режиме

Если ты окажешься в ситуации, когда доступ к интернету ограничен, Comet ML предоставляет возможность вести журнал в автономном режиме. Ты можешь установить COMET_MODE переменную окружения на "offline", чтобы включить эту функцию. Данные твоего эксперимента будут сохранены локально в директории, которую ты сможешь позже загрузить в Comet ML при наличии подключения к интернету.

import os

os.environ["COMET_MODE"] = "offline"

Резюме

Это руководство провело тебя через интеграцию Comet ML с Ultralytics's YOLOv8. Начиная с установки и заканчивая настройкой, ты научился оптимизировать управление экспериментами, получать информацию в реальном времени и адаптировать ведение журнала под нужды твоего проекта.

Изучи официальную документациюComet ML, чтобы узнать больше об интеграции с YOLOv8.

Кроме того, если ты хочешь глубже погрузиться в практическое применение YOLOv8, в частности, для задач сегментации изображений, то это подробное руководство по тонкой настройке YOLOv8 с помощью Comet ML предлагает ценные идеи и пошаговые инструкции для повышения производительности твоей модели.

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

ВОПРОСЫ И ОТВЕТЫ

Как интегрировать Comet ML с Ultralytics YOLOv8 для обучения?

Чтобы интегрировать Comet ML с Ultralytics YOLOv8 , выполни следующие шаги:

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

   pip install ultralytics comet_ml torch torchvision
   

2. Настрой свой API-ключ Comet :

   export COMET_API_KEY=<Your API Key>
   

3. Инициализируй проект Comet в своем коде Python :

   import comet_ml
   
   comet_ml.login(project_name="comet-example-yolov8-coco128")
   

4. Обучи свою модель YOLOv8 и регистрируй метрики:

   from ultralytics import YOLO
   
   model = YOLO("yolov8n.pt")
   results = model.train(
       data="coco8.yaml",
       project="comet-example-yolov8-coco128",
       batch=32,
       save_period=1,
       save_json=True,
       epochs=3,
   )
   

За более подробными инструкциями обратись к разделу о настройкеComet ML.

В чем преимущества использования Comet ML с YOLOv8?

Интегрировав Ultralytics YOLOv8 с Comet ML, ты сможешь:

• Контролируй результаты в реальном времени: Получай мгновенную обратную связь о результатах тренировок, что позволяет быстро вносить коррективы.
• Записывай в журнал обширные метрики: Автоматически фиксируй такие важные метрики, как mAP, потери, гиперпараметры и контрольные точки модели.
• Отслеживай эксперименты в автономном режиме: Веди локальный журнал тренировочных запусков, когда доступ к интернету недоступен.
• Сравнивай разные тренировочные прогоны: Используй интерактивную панель Comet ML dashboard для анализа и сравнения нескольких экспериментов.

Используя эти возможности, ты сможешь оптимизировать свои рабочие процессы машинного обучения для повышения производительности и воспроизводимости. Для получения дополнительной информации посети руководство по интеграцииComet ML.

Как настроить поведение журнала Comet ML во время тренировки YOLOv8 ?

Comet ML позволяет широко настраивать поведение журнала с помощью переменных окружения:

• Измени количество прогнозов изображения в журнале:

  import os
  
  os.environ["COMET_MAX_IMAGE_PREDICTIONS"] = "200"
  

• Настройте интервал регистрации партий:

  import os
  
  os.environ["COMET_EVAL_BATCH_LOGGING_INTERVAL"] = "4"
  

• Отключи ведение журнала матрицы путаницы:

  import os
  
  os.environ["COMET_EVAL_LOG_CONFUSION_MATRIX"] = "false"
  

О дополнительных возможностях настройки читай в разделе " Настройка Comet ML Logging ".

Как просмотреть подробные метрики и визуализации моих тренировок YOLOv8 на Comet ML?

Как только твоя модель YOLOv8 начнет обучение, ты сможешь получить доступ к широкому спектру метрик и визуализаций на панели Comet ML. Основные возможности включают в себя:

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

Подробный обзор этих возможностей ты найдешь в разделе " Понимание эффективности твоей модели с помощью Comet ML-визуализаций ".

Можно ли использовать Comet ML для оффлайн-логирования при обучении моделей YOLOv8 ?

Да, ты можешь включить автономное ведение журнала в Comet ML, установив COMET_MODE Переменная окружения имеет значение "offline":

import os

os.environ["COMET_MODE"] = "offline"

Эта функция позволяет тебе локально регистрировать данные экспериментов, которые позже можно загрузить на сайт Comet ML при наличии подключения к интернету. Это особенно полезно при работе в условиях ограниченного доступа к интернету. Более подробную информацию ты найдешь в разделе " Offline Logging ".

📅 Создано 9 месяцев назад ✏️ Обновлено 3 дня назад

Комментарии