Экспорт настраиваемой модели Ultralytics YOLOv8

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

Обучение пользовательской модели обнаружения объектов

В нашем предыдущем видео мы изучили процесс обучения модели YOLOv8 на пользовательском наборе данных cups в Google Colab. Мы увидели, что график обучения неуклонно улучшается: потери уменьшаются, а среднее значение ошибки растет. Исходя из этого, наша специальная модель теперь может очень точно идентифицировать пять различных типов чашек.

‍

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

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

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

Экспорт и запуск инференса с обученной моделью

Загрузив из Colab обученные веса модели, мы можем без проблем импортировать их в среду Python , готовые раскрыть весь потенциал нашей пользовательской модели.

С помощью нескольких строк кода мы можем настроить Python для запуска живых выводов на веб-камере, захватывая видео в реальном времени и обнаруживая различные чашки с впечатляющей точностью. Сила YOLOv8 проявляется в том, что наша модель без труда определяет чашки различных форм, размеров и цветов, демонстрируя свою универсальность и надежность в реальных условиях.

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

В заключение

От обучения пользовательской модели до экспорта обученных весов и проведения прямых вычислений с помощью веб-камеры - мы убедились в мощности и универсальности YOLOv8 на собственном опыте.

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

Если вам интересно узнать, как экспортировать натренированные модели YOLOv8 и проводить исследования с помощью веб-камеры, посмотрите полное видео здесь!