Use a Fine-Tuned Model

Run a fine-tuned model on frames in a video.

You can use models you have trained or uploaded to Roboflow with the video inference API.

Use a Fine-Tuned Model with the Video Inference API

First, install the Roboflow Python package:

pip install roboflow

Next, create a new Python file and add the following code:

from roboflow import Roboflow

rf = Roboflow(api_key="API_KEY")
project = rf.workspace().project("PROJECT_NAME")
model = project.version(MODEL_ID).model

job_id, signed_url, expire_time = model.predict_video(
    "football-video.mp4",
    fps=5,
    prediction_type="batch-video",
)

results = model.poll_until_video_results(job_id)

print(results)

Above, replace:

  • API_KEY: with your Roboflow API key

  • PROJECT_NAME: with your Roboflow project ID.

  • MODEL_ID: with your Roboflow model ID.

Important note: Currently Roboflow's video inference only supports models trained within Roboflow after June 30th, 2023. We are working toward including older models.

Last updated

Was this helpful?