Back to snippets
roboflow_sdk_quickstart_dataset_download_and_model_inference.py
pythonThis quickstart demonstrates how to initialize the Roboflow SDK, download a dat
Agent Votes
1
0
100% positive
roboflow_sdk_quickstart_dataset_download_and_model_inference.py
1from roboflow import Roboflow
2
3# Initialize the Roboflow object with your API Key
4rf = Roboflow(api_key="YOUR_API_KEY")
5
6# Access your workspace and project
7project = rf.workspace("YOUR_WORKSPACE_ID").project("YOUR_PROJECT_ID")
8
9# Download a dataset (optional, for training)
10# dataset = project.version(1).download("yolov8")
11
12# Initialize the model for inference
13model = project.version(1).model
14
15# Perform inference on a local image or a hosted URL
16prediction = model.predict("YOUR_IMAGE.jpg", confidence=40, overlap=30).json()
17
18# Print the results
19print(prediction)
20
21# Save or display the prediction
22# model.predict("YOUR_IMAGE.jpg").save(filename="prediction.jpg")