trainYOLO
  • trainYOLO platform
  • Getting started
    • 1. Create a new project
    • 2. Upload images
    • 3. Label images
    • 4. Train model
    • 5. Model-assisted labeling
  • CLI
    • Installation
    • Create project
    • Upload data
    • Export data
    • Upload model
  • SDK
    • Installation
    • Create project
    • Upload data
    • Export data
    • Upload model
    • Upload predictions
Powered by GitBook
On this page
  1. SDK

Upload model

You can upload a trained model to your project, to be used for model-assisted labeling, or just to keep track of your different model versions.

from trainyolo.client import Client, Project
from trainyolo.utils.yolov8 import upload_yolov8_run
from trainyolo.utils.yolov5 import upload_yolov5_run

APIKEY = "YOUR_API_KEY"

# get client
client = Client(APIKEY)

# get project by name
name = 'project name'
project = Project.get_by_name(client, name)

# upload yolov8 detection model
upload_yolov8_run(project, mode='detect', iou=0.45)

# upload yolov8 segmentation model
upload_yolov8_run(project, mode='segment', iou=0.45)

# upload yolov5 run
upload_yolov5_run(project, iou=0.45)
PreviousExport dataNextUpload predictions

Last updated 2 years ago