Skip to main content

Official Python package for working with the Roboflow API

Project description

Roboflow Python Package

Roboflow provides everything you need to build and deploy computer vision models. roboflow-python is the official Roboflow Python package. roboflow-python enables you to interact with models, datasets, and projects hosted on Roboflow.

With this Python package, you can:

  1. Create and manage projects;
  2. Upload images, annotations, and datasets to manage in Roboflow;
  3. Start training vision models on Roboflow;
  4. Run inference on models hosted on Roboflow, or Roboflow models self-hosted via Roboflow Inference, and more.

The Python package is documented on the official Roboflow documentation site. If you are developing a feature for this Python package, or need a full Python library reference, refer to the package developer documentation.

💻 Installation

You will need to have Python 3.8 or higher set up to use the Roboflow Python package.

Run the following command to install the Roboflow Python package:

pip install roboflow

For desktop features, use:

pip install "roboflow[desktop]"
Install from source

You can also install the Roboflow Python package from source using the following commands:

git clone https://github.com/roboflow-ai/roboflow-python.git
cd roboflow-python
python3 -m venv env
source env/bin/activate
pip install .
Command line tool

By installing roboflow python package you can use some of its functionality in the command line (without having to write python code). See CLI-COMMANDS.md

🚀 Getting Started

To use the Roboflow Python package, you first need to authenticate with your Roboflow account. You can do this by running the following command:

import roboflow
roboflow.login()
Authenticate with an API key

You can also authenticate with an API key by using the following code:

import roboflow

rf = roboflow.Roboflow(api_key="")

Learn how to retrieve your Roboflow API key.

Quickstart

Below are some common methods used with the Roboflow Python package, presented concisely for reference. For a full library reference, refer to the Roboflow API reference documentation.

import roboflow

roboflow.login()

rf = roboflow.Roboflow()

# create a project
rf.create_project(
    project_name="project name",
    project_type="project-type",
    license="project-license" # "private" for private projects
)

workspace = rf.workspace("WORKSPACE_URL")
project = workspace.project("PROJECT_URL")
version = project.version("VERSION_NUMBER")

# upload a dataset
workspace.upload_dataset(
    dataset_path="./dataset/",
    num_workers=10,
    dataset_format="yolov8", # supports yolov8, yolov5, and Pascal VOC
    project_license="MIT",
    project_type="object-detection"
)

# upload model weights
version.deploy(model_type="yolov8", model_path=f{HOME}/runs/detect/train/)

# upload model weights - yolov10
# Before attempting to upload YOLOv10 models install ultralytics like this:
# pip install git+https://github.com/THU-MIG/yolov10.git
version.deploy(model_type="yolov10", model_path=f{HOME}/runs/detect/train/, filename="weights.pt")

# run inference
model = version.model

img_url = "https://media.roboflow.com/quickstart/aerial_drone.jpeg"

predictions = model.predict(img_url, hosted=True).json()

print(predictions)

Library Structure

The Roboflow Python library is structured using the same Workspace, Project, and Version ontology that you will see in the Roboflow application.

import roboflow

roboflow.login()

rf = roboflow.Roboflow()

workspace = rf.workspace("WORKSPACE_URL")
project = workspace.project("PROJECT_URL")
version = project.version("VERSION_NUMBER")

The workspace, project, and version parameters are the same as those you will find in the URL addresses at app.roboflow.com and universe.roboflow.com.

Within the workspace object you can perform actions like making a new project, listing your projects, or performing active learning where you are using predictions from one project's model to upload images to a new project.

Within the project object, you can retrieve metadata about the project, list versions, generate a new dataset version with preprocessing and augmentation settings, train a model in your project, and upload images and annotations to your project.

Within the version object, you can download the dataset version in any model format, train the version on Roboflow, and deploy your own external model to Roboflow.

🏆 Contributing

We would love your input on how we can improve the Roboflow Python package! Please see our contributing guide to get started. Thank you 🙏 to all our contributors!


Project details


Release history Release notifications | RSS feed

This version

1.2.1

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

roboflow-1.2.1.tar.gz (79.5 kB view details)

Uploaded Source

Built Distribution

roboflow-1.2.1-py3-none-any.whl (86.9 kB view details)

Uploaded Python 3

File details

Details for the file roboflow-1.2.1.tar.gz.

File metadata

  • Download URL: roboflow-1.2.1.tar.gz
  • Upload date:
  • Size: 79.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.8.18

File hashes

Hashes for roboflow-1.2.1.tar.gz
Algorithm Hash digest
SHA256 bcc6fe6067c614dda406902a60aa29e6680fcd1180b943f9c90cf108755175c5
MD5 6782162ea44881b5175cb10fb55f62e3
BLAKE2b-256 75a4f507b248379f4f817f6be80fb526ae736e9f3c357a886898c183e0083979

See more details on using hashes here.

File details

Details for the file roboflow-1.2.1-py3-none-any.whl.

File metadata

  • Download URL: roboflow-1.2.1-py3-none-any.whl
  • Upload date:
  • Size: 86.9 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.8.18

File hashes

Hashes for roboflow-1.2.1-py3-none-any.whl
Algorithm Hash digest
SHA256 755c17605155d7049e936283a6339713403e8ecc8049f5b38df633541c9ce38b
MD5 e767f85468e8d7c7bd0be3ca8177bc59
BLAKE2b-256 64884d093e161340db475573f82d8417ada9c6eed8ab66ad5e7cbc86ca7ed541

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page