HUB Integrations
🚧 Under Construction 🚧
Welcome to the Integrations guide for Ultralytics HUB! We are in the process of expanding this section to provide you with comprehensive guidance on integrating your YOLOv5 and YOLOv8 models with various platforms and formats. Currently, Roboflow is our available dataset integration, with a wide range of export integrations for your trained models.
Watch: Train Your Custom YOLO Models In A Few Clicks with Ultralytics HUB.
Available Integrations
Dataset Integrations
- Roboflow: Seamlessly import your datasets for training.
Export Integrations
Available export formats are in the table below. You can predict or validate directly on exported models using the ultralytics Python package, i.e. yolo predict model=yolov8n.onnx.
| Format | format Argument |
Model | Metadata | Arguments |
|---|---|---|---|---|
| PyTorch | - | yolov8n.pt |
✅ | - |
| TorchScript | torchscript |
yolov8n.torchscript |
✅ | imgsz, optimize, batch |
| ONNX | onnx |
yolov8n.onnx |
✅ | imgsz, half, dynamic, simplify, opset, batch |
| OpenVINO | openvino |
yolov8n_openvino_model/ |
✅ | imgsz, half, int8, batch |
| TensorRT | engine |
yolov8n.engine |
✅ | imgsz, half, dynamic, simplify, workspace, int8, batch |
| CoreML | coreml |
yolov8n.mlpackage |
✅ | imgsz, half, int8, nms, batch |
| TF SavedModel | saved_model |
yolov8n_saved_model/ |
✅ | imgsz, keras, int8, batch |
| TF GraphDef | pb |
yolov8n.pb |
❌ | imgsz, batch |
| TF Lite | tflite |
yolov8n.tflite |
✅ | imgsz, half, int8, batch |
| TF Edge TPU | edgetpu |
yolov8n_edgetpu.tflite |
✅ | imgsz, batch |
| TF.js | tfjs |
yolov8n_web_model/ |
✅ | imgsz, half, int8, batch |
| PaddlePaddle | paddle |
yolov8n_paddle_model/ |
✅ | imgsz, batch |
| NCNN | ncnn |
yolov8n_ncnn_model/ |
✅ | imgsz, half, batch |
Coming Soon
- Additional Dataset Integrations
- Detailed Export Integration Guides
- Step-by-Step Tutorials for Each Integration
Need Immediate Assistance?
While we're in the process of creating detailed guides:
- Browse through other HUB Docs for detailed guides and tutorials.
- Raise an issue on our GitHub for technical support.
- Join our Discord Community for live discussions and community support.
We appreciate your patience as we work to make this section comprehensive and user-friendly. Stay tuned for updates!