Key Benefit: Train your model entirely within a Google Colab notebook environment, eliminating the need for a local NVIDIA GPU or x86 processor.
This Colab notebook, dataset_yolo_onnx.ipynb, guides you in 6 steps: prepare your dataset, train a YOLO11 model, and convert it to an ONNX file (used for part 2).
- Prepare Files - Download a dataset in yolo format or create your own.
- Zip and Upload - Zip your dataset and upload it to your Google Drive.
- Colab Access - Unzip the dataset in your Colab notebook environment.
- Train YOLO11 - Use the provided code to train a YOLO11 model.
- Convert to ONNX - Convert the trained model best.pt to ONNX format best.onnx for Hailo's Dataflow Compiler.
- Save model - Zip training folders with the best.pt and best.onnx in the folder weights.