Links
Comment on page

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 modified 22d ago