8.9 KiB
comments | description | keywords |
---|---|---|
true | Исследуйте разнообразный спектр поддерживаемых Ultralytics моделей семейства YOLO, SAM, MobileSAM, FastSAM, YOLO-NAS и RT-DETR. Начните работу с примерами использования как для CLI, так и для Python. | Ultralytics, документация, YOLO, SAM, MobileSAM, FastSAM, YOLO-NAS, RT-DETR, модели, архитектуры, Python, CLI |
Поддерживаемые модели Ultralytics
Добро пожаловать в документацию по моделям Ultralytics! Мы поддерживаем широкий спектр моделей, каждая из которых адаптирована для конкретных задач, таких как обнаружение объектов, сегментация на уровне инстанций, классификация изображений, оценка позы и слежение за несколькими объектами. Если вы заинтересованы в добавлении архитектуры своей модели в Ultralytics, ознакомьтесь с нашим Руководством для участников.
!!! Note "Примечание"
🚧 Наша документация на разных языках находится в стадии разработки, и мы усердно работаем над ее улучшением. Спасибо за ваше терпение! 🙏
Особенные модели
Вот некоторые ключевые поддерживаемые модели:
- YOLOv3: Третье поколение семейства моделей YOLO, авторства Джозефа Редмона, известное своей эффективностью в реальном времени для обнаружения объектов.
- YOLOv4: Нативное для darknet обновление YOLOv3, выпущенное Алексеем Бочковским в 2020 году.
- YOLOv5: Улучшенная версия архитектуры YOLO от Ultralytics, предлагающая лучшие компромиссы производительности и скорости по сравнению с предыдущими версиями.
- YOLOv6: Выпущенная в 2022 году компанией Meituan и используемая во многих роботах автономной доставки компании.
- YOLOv7: Обновленные модели YOLO, выпущенные в 2022 году авторами YOLOv4.
- YOLOv8 НОВИНКА 🚀: Последняя версия семейства YOLO, обладающая расширенными возможностями, такими как сегментация на уровне инстанций, оценка позы/ключевых точек и классификация.
- Segment Anything Model (SAM): Модель сегментации всего и вся (SAM) от Meta.
- Mobile Segment Anything Model (MobileSAM): MobileSAM для мобильных приложений от университета Kyung Hee.
- Fast Segment Anything Model (FastSAM): FastSAM от Группы анализа изображений и видео, Института автоматики, Китайской академии наук.
- YOLO-NAS: Модели нейронной архитектуры поиска YOLO (NAS).
- Realtime Detection Transformers (RT-DETR): Модели трансформеров реального времени для обнаружения объектов (RT-DETR) от Baidu PaddlePaddle.
Смотрите: Запуск моделей YOLO от Ultralytics всего в несколько строк кода.
Начало работы: Примеры использования
Этот пример предоставляет простые примеры обучения и вывода для YOLO. Полная документация по этим и другим режимам представлена на страницах документации Predict, Train, Val и Export.
Обратите внимание, что ниже приведен пример для моделей Detect YOLOv8 для обнаружения объектов. Для дополнительных поддерживаемых задач смотрите документацию по Segment, Classify и Pose.
!!! Example "Пример"
=== "Python"
Предобученные модели PyTorch `*.pt`, а также конфигурационные файлы `*.yaml` могут быть переданы в классы `YOLO()`, `SAM()`, `NAS()` и `RTDETR()`, чтобы создать экземпляр модели на Python:
```python
from ultralytics import YOLO
# Загрузить предварительно обученную модель YOLOv8n для COCO
model = YOLO('yolov8n.pt')
# Отобразить информацию о модели (необязательно)
model.info()
# Обучить модель на примерном наборе данных COCO8 в течение 100 эпох
results = model.train(data='coco8.yaml', epochs=100, imgsz=640)
# Запустить вывод с помощью модели YOLOv8n на изображении 'bus.jpg'
results = model('path/to/bus.jpg')
```
=== "CLI"
Команды CLI доступны для непосредственного запуска моделей:
```bash
# Загрузить предварительно обученную модель YOLOv8n для COCO и обучить её на примерном наборе данных COCO8 в течение 100 эпох
yolo train model=yolov8n.pt data=coco8.yaml epochs=100 imgsz=640
# Загрузить предварительно обученную модель YOLOv8n для COCO и запустить вывод на изображении 'bus.jpg'
yolo predict model=yolov8n.pt source=path/to/bus.jpg
```
Вклад в новые модели
Заинтересованы в том, чтобы внести свою модель в Ultralytics? Отлично! Мы всегда открыты для расширения нашего портфолио моделей.
-
Сделайте Fork Репозитория: Начните с создания форка репозитория Ultralytics на GitHub.
-
Склонируйте свой Fork: Склонируйте ваш форк на локальную машину и создайте новую ветку для работы.
-
Реализуйте свою Модель: Добавьте вашу модель, следуя стандартам программирования и руководящим принципам, указанным в нашем Руководстве для участников.
-
Тщательно протестируйте: Убедитесь, что вы тщательно протестировали свою модель, как изолированно, так и как часть пайплайна.
-
Создайте Pull Request: Как только вы будете удовлетворены своей моделью, создайте pull request в основной репозиторий для рассмотрения.
-
Код-ревью и Слияние: После рассмотрения, если ваша модель соответствует нашим критериям, она будет объединена с основным репозиторием.
Для подробных инструкций см. наше Руководство для участников.