Настраивайтесь на YOLO Vision 2025!
25 сентября 2025 г.
10:00 — 18:00 BST
Гибридное мероприятие
Yolo Vision 2024

Как обучать собственные модели с помощью Ultralytics HUB

Nuvola Ladi

3 мин чтения

31 мая 2024 г.

Узнайте, как легко обучать пользовательские модели YOLO с помощью Ultralytics HUB. Это пошаговое руководство по интуитивно понятной платформе предлагает обзор бесшовной интеграции, отслеживания в реальном времени и облачного обучения.

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

Что такое Ultralytics HUB?

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

Он интегрируется с различными платформами, поддерживает обновления в реальном времени через мобильное приложение (доступно как на Android, так и на iOS) и постоянно развивается благодаря новым функциям. Для получения более подробной технической информации и пошагового рабочего процесса ознакомьтесь с нашей статьей на Medium о том, как обучать модели компьютерного зрения в облаке.

Начало работы с Ultralytics HUB

Удобный интерфейс

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

Наборы данных и модели

Ultralytics HUB поставляется с предварительно загруженными стандартными наборами данных, такими как VOC, COCO и Simpsons, которые обычно используются для сравнительного анализа. Эти наборы данных являются существенными: COCO содержит 80 классов и 140 000 изображений. Однако платформа также поддерживает наборы данных меньшего размера, что делает ее идеальной для пользователей всех уровней. Вы можете загрузить свой собственный набор данных или подключиться к внешним инструментам, таким как Roboflow, для аннотации изображений и подготовки набора данных.

Создание проекта

Создание нового проекта в Ultralytics HUB невероятно просто. Вот пошаговое руководство, которое поможет вам начать:

  1. Создайте проект: начните с названия вашего проекта. Для этого примера давайте назовем его «Object Detection».
  2. Выберите набор данных: выберите набор данных из доступных вариантов. Для простоты мы будем использовать набор данных Simpsons с 14 000 изображений.
  3. Выберите модель: выберите модель YOLO. Ultralytics HUB поддерживает различные модели, включая YOLOv5 и YOLOv8. Для целей данного примера мы выберем нано-модель YOLOv8.
  4. Настройка гиперпараметров: Настройте такие параметры, как количество эпох, размер изображения и размер пакета. Вы также можете выбрать, использовать ли GPU или CPU, и настроить параметры кэширования.

Обучение модели

После настройки проекта, обучение модели начинается одним кликом. Ultralytics HUB предлагает несколько вариантов обучения:

  • Локальное обучение: установите Ultralytics на свой локальный компьютер и запустите скрипт обучения.
  • Google Colab: Откройте блокнот Google Colab с предварительно настроенным кодом. Этот вариант не требует программирования и безупречно работает в облаке.
  • HUB Cloud: Облачная платформа Ultralytics HUB Training предлагает решение для обучения моделей YOLO без написания кода, идеально подходящее для тех, кто не занимается программированием, и владельцев бизнеса. Рабочий процесс включает загрузку набора данных, выбор модели и настройку облачного экземпляра, что упрощает точную настройку предварительно обученных моделей и их экспорт для различных приложений.

Чтобы обучить модель в Google Colab:

  1. Установите Ultralytics: выполните команду для установки Ultralytics в блокноте.
  2. Настройка и ключ API: Выполните настройку и введите свой ключ API.
  3. Начать обучение: Выполните команды обучения, и ваша модель начнет тренироваться.
__wf_reserved_inherit
Рис. 1. Николай Нильсен рассказывает о том, как обучать модели на Ultralytics HUB.

Мониторинг прогресса обучения

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

Расширенные функции и интеграции

Облачное обучение и интеграция с RoboFlow

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

Мобильное приложение

Приложение Ultralytics HUB обеспечивает обнаружение объектов в реальном времени на вашем мобильном устройстве. Используя предварительно обученные модели, приложение может обнаруживать объекты из набора данных COCO с высокой точностью, работая со скоростью 30 кадров в секунду на iPhone 14 Pro. Эта функция идеально подходит для тестирования моделей в реальных сценариях и демонстрации возможностей ваших обученных моделей.

В заключение

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

Так зачем ждать? Погрузитесь в Ultralytics HUB сегодня и раскройте потенциал своих проектов машинного обучения всего несколькими щелчками мыши!

Давайте строить будущее
ИИ вместе!

Начните свой путь в будущее машинного обучения

Начать бесплатно
Ссылка скопирована в буфер обмена