Кастомное обучение Ultralytics YOLO11 для оценки позы собаки

Что если осанка твоей собаки могла бы подсказать тебе, как она себя чувствует? Следить за ней круглые сутки вручную не так-то просто. Однако благодаря достижениям в области искусственного интеллекта (ИИ) и computer vision мы можем анализировать видеозаписи в реальном времени, чтобы лучше понимать её поведение.

В частности, модели computer vision, такие как Ultralytics YOLO11, могут помочь оценить позу собаки и отслеживать её движения, предоставляя ценные данные о её самочувствии. Как это работает? Функция pose estimation в YOLO11 может определять ключевые точки на теле объекта, чтобы понять его позу и движения.

Кроме того, YOLO11 можно дообучить на датасете, предназначенном для оценки позы собак, что позволяет точно анализировать язык тела твоего питомца. Пакет Python от Ultralytics поддерживает Dog-Pose Dataset, что упрощает обучение и развертывание моделей vision AI для собак. Эта технология является частью быстро растущего pet tech market, объем которого в 2024 году составил 9,4 миллиарда долларов, а к 2037 году, по прогнозам, достигнет 64 миллиардов долларов.

Вдохновением для этой статьи послужил Блюз, наш главный директор по собакам (DEO). Если ты заглянешь на нашу страницу About Us, то увидишь, что Блюз — ценный член команды и играет важную роль в поддержании веселой атмосферы в Ultralytics!

В этой статье мы обсудим, как обучить Ultralytics YOLO11 на собственных данных с помощью Dog-Pose Dataset для оценки позы собак. Мы также рассмотрим практические варианты применения в уходе за домашними животными и анализе их поведения.

Рис. 1. Знакомься, Блюз, собачий директор (DEO) в Ultralytics.

Обзор набора данных Dog-Pose#

Dataset — это набор данных, используемый для обучения моделей machine learning. Для pose estimation идеальный датасет включает изображения с размеченными ключевыми точками для отображения положений тела. В нем также должны быть представлены различные позы, ракурсы, условия освещения и фоны, чтобы помочь модели научиться точно распознавать и предсказывать позы. Такое разнообразие делает модель более надежной для использования в реальных условиях.

Набор данных Dog-Pose, поддерживаемый Ultralytics, специально разработан для того, чтобы помочь моделям эффективно изучать и распознавать позы собак. Он включает более 8400 аннотированных изображений различных пород собак с детальной разметкой 24 ключевых точек, таких как хвост, уши и лапы.

Fig 2. Обзор Dog-Pose Dataset.

Как обучить Ultralytics YOLO11 на собственных данных с помощью набора данных Dog-Pose Dataset#

Кастомное обучение YOLO11 на Dog-Pose Dataset — это простой процесс. Чтобы начать, тебе нужно настроить свое окружение, установив Ultralytics Python package, который включает все необходимые инструменты для обучения и оценки.

Ultralytics имеет встроенную поддержку Dog-Pose Dataset, что исключает необходимость ручной разметки и позволяет сразу приступить к обучению. Как только все настроено, ты сможешь обучить YOLO11 на Dog-Pose Dataset с помощью всего нескольких строк кода, как показано на изображении ниже.

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

Рис. 3. Пример кода, демонстрирующий пользовательское обучение Ultralytics YOLO11 на наборе данных Dog-Pose Dataset.

Если при обучении модели ты столкнешься с какими-либо проблемами, вот несколько советов по устранению неполадок, которые помогут тебе быстро их решить:

• Проверь подключение к интернету: Dog-Pose Dataset загружается автоматически при запуске скрипта обучения. Убедись, что у тебя стабильное интернет-соединение, чтобы избежать сбоев при загрузке.
• Обнови Ultralytics: Убедись, что используешь последнюю версию пакета Ultralytics Python.
• Проверь наличие ошибок в консоли: Внимательно читай любые сообщения об ошибках, так как они часто дают подсказки о том, что нужно исправить.

Ты можешь ознакомиться с Ultralytics Common Issues Guide для получения дополнительных советов по устранению неполадок.

Что происходит во время обучения модели?#

Возможно, тебе интересно, что происходит «под капотом», когда ты выполняешь custom-train Ultralytics YOLO11 на наборе данных Dog-Pose Dataset. Давай взглянем на этот процесс поближе.

Вместо того чтобы начинать с нуля, мы используем предобученную модель Ultralytics YOLO11-pose, которая уже обучена на датасете COCO-Pose dataset. Эта предобученная модель может обнаруживать ключевые точки человека, так как COCO-Pose предназначена для оценки позы человека. Фактически, без какого-либо дополнительного обучения ты можешь использовать YOLO11-Pose для оценки позы человека прямо "из коробки".

Благодаря переносу обучения (transfer learning) мы адаптируем эту модель специально для оценки позы собаки, помогая ей распознавать ключевые точки, такие как лапы, хвост и голова. Предоставляя модели примеры именно с собаками, она учится фокусироваться на этих важных признаках.

Во время обучения некоторые части модели остаются неизменными, сохраняя общие знания, полученные из набора данных COCO. Другие части дообучаются для повышения точности оценки поз собак. Модель учится, сравнивая свои предсказания с фактическими ключевыми точками в наборе данных и корректируя их для уменьшения ошибок. Со временем этот процесс улучшает способность модели точно отслеживать движения собаки.

Transfer learning также позволяет модели адаптироваться к разным породам, размерам и паттернам движения, обеспечивая ее надежную работу в реальных сценариях.

Рис. 4. Определение ключевых точек у Блюза и его сестры Хэппи.

Преимущества Ultralytics YOLO11 в задаче оценки позы собаки#

Существует множество различных computer vision models, так почему же именно YOLO11 является правильным выбором для оценки позы собаки?

Ultralytics YOLO11 выделяется своей скоростью в реальном времени и точностью, что делает его отличным вариантом для оценки позы собаки. Он превосходит предыдущие версии как по точности, имея на 22% меньше параметров, чем YOLOv8, так и по скорости, достигая более высокого среднего значения средней точности (mAP) на датасете COCO, что означает более точное и эффективное обнаружение объектов. Его высокая скорость обработки делает его идеальным для приложений реального времени, где быстрая и надежная детекция имеет решающее значение.

Помимо оценки позы, Ultralytics YOLO11 также поддерживает такие computer vision tasks, как сегментация экземпляров и трекинг объектов, которые помогают создать более комплексное решение на базе искусственного интеллекта для наблюдения за твоей собакой. Эти функции могут улучшить отслеживание движений, анализ поведения и общий уход за питомцем.

Рис. 5. Ultralytics YOLO11 в действии: простая и эффективная сегментация собаки по кличке Блюз!

Применение оценки позы собак и Ultralytics YOLO11#

Далее давай обсудим практическое применение оценки позы собаки и её влияние на уход за питомцами.

Улучшение дрессировки питомца с помощью оценки позы#

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

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

Например, представь, что ты учишь собаку команде «сидеть». Система может отслеживать позу собаки и определять, полностью ли она села. Если собака опускает заднюю часть тела, но не садится полностью, система распознает неполное действие и отправит мгновенную обратную связь через приложение. Дрессировщик получит сигнал сделать небольшие корректировки в тренировке, например, повторить команду или направить собаку в правильное положение.

Рис. 6. Блюз помогает команде с оценкой позы собак с помощью Ultralytics YOLO11

Развитие ветеринарного ухода с помощью моделей оценки позы#

Computer vision может изменить подход ветеринаров к уходу за животными. Способность оценки позы собаки точно анализировать детали позволяет легче замечать необычные модели движений и выявлять потенциальные проблемы со здоровьем.

Например, ветеринар, наблюдающий за собакой после травмы связок, может положиться на автоматизированный анализ с помощью Ultralytics YOLO11, обученного на Dog-Pose Dataset. Хромоту или изменения в положении лап можно легко обнаружить. Непрерывный мониторинг 24/7 с поддержкой компьютерного зрения дает четкое представление о процессе восстановления собаки, помогая определить, работает ли лечение или нужны корректировки.

Будущее оценки позы собаки#

По мере развития технологий такие решения, как оценка позы собаки с помощью YOLO11, вероятно, будут играть большую роль в animal monitoring и обеспечении благополучия. Фактически, YOLO11 можно интегрировать с носимыми технологиями, такими как умные ошейники и трекеры здоровья, для мониторинга ключевых показателей здоровья, таких как частота сердечных сокращений, уровень активности и паттерны подвижности.

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

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

Основные выводы#

Благодаря таким инновациям, как Ultralytics YOLO11 и Dog-Pose Dataset, мы открываем новые возможности в компьютерном зрении. Эти достижения помогают нам лучше понимать поведение и здоровье собак так, как раньше было невозможно.

Точно отслеживая позы собак, мы можем улучшить дрессировку, контролировать здоровье и сделать уход за питомцами более эффективным. Будь то исследования, ветеринарная помощь или дрессировка, vision AI создает более умные способы заботы о наших собаках и улучшения их самочувствия.

Присоединяйся к our community и изучи our GitHub repository, чтобы узнать больше об ИИ. Узнай, как такие инновации, как computer vision in healthcare и AI in self-driving cars, формируют будущее. Ознакомься с нашими licensing options, чтобы начать свои проекты в области vision AI уже сегодня.