YOLOv8 🚀 on AzureML

What is Azure?

Azure is Microsoft's cloud computing platform, designed to help organizations move their workloads to the cloud from on-premises data centers. With the full spectrum of cloud services including those for computing, databases, analytics, machine learning, and networking, users can pick and choose from these services to develop and scale new applications, or run existing applications, in the public cloud.

What is Azure Machine Learning (AzureML)?

Azure Machine Learning, commonly referred to as AzureML, is a fully managed cloud service that enables data scientists and developers to efficiently embed predictive analytics into their applications, helping organizations use massive data sets and bring all the benefits of the cloud to machine learning. AzureML offers a variety of services and capabilities aimed at making machine learning accessible, easy to use, and scalable. It provides capabilities like automated machine learning, drag-and-drop model training, as well as a robust Python SDK so that developers can make the most out of their machine learning models.

How Does AzureML Benefit YOLO Users?

For users of YOLO (You Only Look Once), AzureML provides a robust, scalable, and efficient platform to both train and deploy machine learning models. Whether you are looking to run quick prototypes or scale up to handle more extensive data, AzureML's flexible and user-friendly environment offers various tools and services to fit your needs. You can leverage AzureML to:

• Easily manage large datasets and computational resources for training.
• Utilize built-in tools for data preprocessing, feature selection, and model training.
• Collaborate more efficiently with capabilities for MLOps (Machine Learning Operations), including but not limited to monitoring, auditing, and versioning of models and data.

In the subsequent sections, you will find a quickstart guide detailing how to run YOLOv8 object detection models using AzureML, either from a compute terminal or a notebook.

Prerequisites

Before you can get started, make sure you have access to an AzureML workspace. If you don't have one, you can create a new AzureML workspace by following Azure's official documentation. This workspace acts as a centralized place to manage all AzureML resources.

Create a compute instance

From your AzureML workspace, select Compute > Compute instances > New, select the instance with the resources you need.

Quickstart from Terminal

Start your compute and open a Terminal:

Create virtualenv

Create your conda virtualenv and install pip in it:

conda create --name yolov8env -y
conda activate yolov8env
conda install pip -y

Install the required dependencies:

cd ultralytics
pip install -r requirements.txt
pip install ultralytics
pip install onnx>=1.12.0

Perform YOLOv8 tasks

Predict:

yolo predict model=yolov8n.pt source='https://ultralytics.com/images/bus.jpg'

Train a detection model for 10 epochs with an initial learning_rate of 0.01:

yolo train data=coco8.yaml model=yolov8n.pt epochs=10 lr0=0.01

You can find more instructions to use the Ultralytics CLI here.

Quickstart from a Notebook

Create a new IPython kernel

Open the compute Terminal.

From your compute terminal, you need to create a new ipykernel that will be used by your notebook to manage your dependencies:

conda create --name yolov8env -y
conda activate yolov8env
conda install pip -y
conda install ipykernel -y
python -m ipykernel install --user --name yolov8env --display-name "yolov8env"

Close your terminal and create a new notebook. From your Notebook, you can select the new kernel.

Then you can open a Notebook cell and install the required dependencies:

%%bash
source activate yolov8env
cd ultralytics
pip install -r requirements.txt
pip install ultralytics
pip install onnx>=1.12.0

Note that we need to use the source activate yolov8env for all the %%bash cells, to make sure that the %%bash cell uses environment we want.

Run some predictions using the Ultralytics CLI:

%%bash
source activate yolov8env
yolo predict model=yolov8n.pt source='https://ultralytics.com/images/bus.jpg'

Or with the Ultralytics Python interface, for example to train the model:

from ultralytics import YOLO

# Load a model
model = YOLO("yolov8n.pt")  # load an official YOLOv8n model

# Use the model
model.train(data="coco8.yaml", epochs=3)  # train the model
metrics = model.val()  # evaluate model performance on the validation set
results = model("https://ultralytics.com/images/bus.jpg")  # predict on an image
path = model.export(format="onnx")  # export the model to ONNX format

You can use either the Ultralytics CLI or Python interface for running YOLOv8 tasks, as described in the terminal section above.

By following these steps, you should be able to get YOLOv8 running quickly on AzureML for quick trials. For more advanced uses, you may refer to the full AzureML documentation linked at the beginning of this guide.

Explore More with AzureML

This guide serves as an introduction to get you up and running with YOLOv8 on AzureML. However, it only scratches the surface of what AzureML can offer. To delve deeper and unlock the full potential of AzureML for your machine learning projects, consider exploring the following resources:

• Create a Data Asset: Learn how to set up and manage your data assets effectively within the AzureML environment.
• Initiate an AzureML Job: Get a comprehensive understanding of how to kickstart your machine learning training jobs on AzureML.
• Register a Model: Familiarize yourself with model management practices including registration, versioning, and deployment.
• Train YOLOv8 with AzureML Python SDK: Explore a step-by-step guide on using the AzureML Python SDK to train your YOLOv8 models.
• Train YOLOv8 with AzureML CLI: Discover how to utilize the command-line interface for streamlined training and management of YOLOv8 models on AzureML.

FAQ

How do I run YOLOv8 on AzureML for model training?

Running YOLOv8 on AzureML for model training involves several steps:

1. Create a Compute Instance: From your AzureML workspace, navigate to Compute > Compute instances > New, and select the required instance.

2. Setup Environment: Start your compute instance, open a terminal, and create a conda environment:

   conda create --name yolov8env -y
   conda activate yolov8env
   conda install pip -y
   pip install ultralytics onnx>=1.12.0
   

3. Run YOLOv8 Tasks: Use the Ultralytics CLI to train your model:

   yolo train data=coco8.yaml model=yolov8n.pt epochs=10 lr0=0.01
   

For more details, you can refer to the instructions to use the Ultralytics CLI.

What are the benefits of using AzureML for YOLOv8 training?

AzureML provides a robust and efficient ecosystem for training YOLOv8 models:

• Scalability: Easily scale your compute resources as your data and model complexity grows.
• MLOps Integration: Utilize features like versioning, monitoring, and auditing to streamline ML operations.
• Collaboration: Share and manage resources within teams, enhancing collaborative workflows.

These advantages make AzureML an ideal platform for projects ranging from quick prototypes to large-scale deployments. For more tips, check out AzureML Jobs.

How do I troubleshoot common issues when running YOLOv8 on AzureML?

Troubleshooting common issues with YOLOv8 on AzureML can involve the following steps:

• Dependency Issues: Ensure all required packages are installed. Refer to the requirements.txt file for dependencies.
• Environment Setup: Verify that your conda environment is correctly activated before running commands.
• Resource Allocation: Make sure your compute instances have sufficient resources to handle the training workload.

For additional guidance, review our YOLO Common Issues documentation.

Can I use both the Ultralytics CLI and Python interface on AzureML?

Yes, AzureML allows you to use both the Ultralytics CLI and the Python interface seamlessly:

• CLI: Ideal for quick tasks and running standard scripts directly from the terminal.

  yolo predict model=yolov8n.pt source='https://ultralytics.com/images/bus.jpg'
  

• Python Interface: Useful for more complex tasks requiring custom coding and integration within notebooks.

  from ultralytics import YOLO
  
  model = YOLO("yolov8n.pt")
  model.train(data="coco8.yaml", epochs=3)
  

Refer to the quickstart guides for more detailed instructions here and here.

What is the advantage of using Ultralytics YOLOv8 over other object detection models?

Ultralytics YOLOv8 offers several unique advantages over competing object detection models:

• Speed: Faster inference and training times compared to models like Faster R-CNN and SSD.
• Accuracy: High accuracy in detection tasks with features like anchor-free design and enhanced augmentation strategies.
• Ease of Use: Intuitive API and CLI for quick setup, making it accessible both to beginners and experts.

To explore more about YOLOv8's features, visit the Ultralytics YOLO page for detailed insights.

Comments