ultralytics / hub

Ultralytics HUB tutorials and support
https://hub.ultralytics.com
GNU Affero General Public License v3.0
138 stars 14 forks source link
artificial-intelligence deep-learning deploy hub machine-learning train ultralytics yolov5 yolov8

[中文](https://docs.ultralytics.com/zh/hub/) | [한국어](https://docs.ultralytics.com/ko/hub/) | [日本語](https://docs.ultralytics.com/ja/hub/) | [Русский](https://docs.ultralytics.com/ru/hub/) | [Deutsch](https://docs.ultralytics.com/de/hub/) | [Français](https://docs.ultralytics.com/fr/hub/) | [Español](https://docs.ultralytics.com/es/hub/) | [Português](https://docs.ultralytics.com/pt/hub/) | [Türkçe](https://docs.ultralytics.com/tr/hub/) | [Tiếng Việt](https://docs.ultralytics.com/vi/hub/) | [العربية](https://docs.ultralytics.com/ar/hub/) CI CPU Open In Colab Discord Ultralytics Forums Ultralytics Reddit 👋 Hello from the [Ultralytics](https://www.ultralytics.com/) Team! We've been working hard these last few months to launch [Ultralytics HUB](https://www.ultralytics.com/hub), a new web tool for training and deploying all your YOLOv5 and YOLOv8 🚀 models from one spot! We hope that the resources here will help you get the most out of HUB. Please browse the HUB Docs for details, raise an issue on GitHub for support, and join our Discord community for questions and discussions!
Ultralytics GitHub Ultralytics LinkedIn Ultralytics Twitter Ultralytics YouTube Ultralytics TikTok Ultralytics BiliBili Ultralytics Discord

🗂️ 1. Upload a Dataset

Ultralytics HUB datasets align with the format used by YOLOv5 and YOLOv8. They adhere to the same structural and labeling conventions, ensuring a seamless experience.

Dataset Preparation:

Ensure that the YAML file describing your dataset is placed in the root directory of your dataset, as illustrated below. Once in place, zip the directory for uploading to Ultralytics HUB. The dataset YAML, its directory, and the zip file should all bear the identical name.

For instance, with a dataset named 'coco8', as shown in ultralytics/hub/example_datasets/coco8.zip, include a coco8.yaml within the coco8/ directory. Zip this to form coco8.zip for upload with the command:

# Zipping the dataset directory for upload
zip -r coco8.zip coco8

Explore the provided example_datasets/coco8.zip for insights on structuring your custom dataset.

YAML Specifications:

The dataset YAML follows the established YOLOv5 and YOLOv8 format. For exhaustive instructions, refer to the Train Custom Data tutorial.

# YAML configuration example for custom datasets
path: # dataset root dir (leave empty for HUB integration)
train: images/train # train images (relative to 'path') 8 images
val: images/val # val images (relative to 'path') 8 images
test: # test images (optional)

# Class labels
names:
  0: person
  1: bicycle
  2: car
  3: motorcycle
  # Add more classes as needed

Upload your zipped dataset to Ultralytics HUB by logging in, navigating to the 'Datasets' tab, and selecting 'Upload Dataset'. This lets you scan and view your dataset prior to training YOLOv5 or YOLOv8 models.

HUB Dataset Upload

🚀 2. Train a Model

Connect to the Ultralytics HUB notebook and employ your model API key to embark on model training!

Open In Colab

🌐 3. Deploy to the Real World

Transition your model to 13 different formats including TensorFlow, ONNX, OpenVINO, CoreML, Paddle, and more. Operate your models directly on your iOS or Android mobile device by downloading the Ultralytics App!

❓ Have Issues or Questions?

For those just embarking on their Ultralytics HUB journey, the Issues tab is your go-to resource for support. Click the 'New Issue' button and share your thoughts or questions. Our aim is to enhance your experience with invaluable solutions and improvements! 😃


Ultralytics GitHub Ultralytics LinkedIn Ultralytics Twitter Ultralytics YouTube Ultralytics TikTok Ultralytics BiliBili Ultralytics Discord