Обнаружение ключевых точек является важным аспектом приложений компьютерного зрения, расширяя возможности таких задач, как оценка позы человека и манипулирование роботами. В этом руководстве мы проведем вас через процесс обучения пользовательской модели обнаружения ключевых точек с использованием модели Ultralytics YOLOv8-pose и платформы trainYOLO. В частности, мы обучим модель обнаруживать маркеры белой доски с отдельными ключевыми точками для верхней и нижней частей. Мы рассмотрим каждый шаг, от маркировки вашего набора данных до использования маркировки с помощью модели, чтобы помочь вам создать точный детектор ключевых точек. Давайте начнем!

Шаг 1: Создайте новый проект

Начните с создания нового проекта на платформе trainYOLO. Войдите в свою учетную запись trainYOLO или создайте бесплатную учетную запись здесь, если у вас ее нет. Нажмите на зеленый значок плюс и установите для видимости проекта значение Общий или Частный, в зависимости от ваших предпочтений. В этом примере мы назовем проект Маркеры белой доски и выберем тип аннотации Ключевые точки. Мы добавляем одну категорию объектов Маркер и включаем две категории ключевых точек: Верх и Низ.

Шаг 2: Загрузите свои изображения

Перейдите на вкладку данных в вашем проекте и нажмите зеленый значок «плюс», чтобы загрузить изображения. Выберите изображения, которые вы хотите использовать для обучения, мы загрузим 123 изображения маркеров для белых досок в различных конфигурациях, и нажмите «Загрузить». Имейте в виду, что для более сложных вариантов использования может потребоваться большее количество изображений для обучения оптимальной модели.

Шаг 3: Назовите свои изображения

Как только ваши изображения загружены, приступайте к маркировке каждого изображения. Нажав на изображение, вы попадаете в редактор маркировки. Используйте инструмент Box из левого меню, чтобы точно пометить каждый объект (в нашем случае каждый маркер). Прежде чем рисовать ограничивающую рамку, убедитесь, что сверху выбрана правильная категория объекта, и нарисуйте рамку вокруг объекта, щелкнув один раз в верхнем левом углу и один раз в правом нижнем углу. Затем пометьте каждую ключевую точку, щелкнув соответствующее место. Используйте клавиши со стрелками или карусель внизу, чтобы перемещаться между изображениями и маркировать их соответствующим образом.

Шаг 4: Обучите модель YOLOv8

После маркировки достаточного количества изображений пришло время обучить вашу пользовательскую модель обнаружения ключевых точек YOLOv8. Для простоты мы будем использовать предварительно настроенные блокноты Google Colab, предоставленные trainYOLO. Поэтому перейдите на вкладку модели в своем проекте и выберите блокнот YOLOv8, нажав зеленый значок «плюс». Этот ноутбук поставляется с предварительно настроенными необходимыми настройками. Введите свой API-ключ (доступен при нажатии на ваш аватар) и название проекта. Оставьте остальные параметры по умолчанию, затем начните тренировку, нажав «Время выполнения» и выбрав «Выполнить все». Блокнот загрузит ваш набор данных, обучит модель и загрузит обученную модель в ваш проект, предоставив вам пользовательскую модель обнаружения ключевых точек в кратчайшие сроки.

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

Для нашего примера мы получили следующие показатели:

Шаг 5: Маркировка с помощью модели

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

Следующие шаги:

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

Первоначально опубликовано на https://www.trainyolo.com/blog 11 мая 2023 г.