Roboflow Python
Roboflow streamlines your computer vision pipeline - upload data, label it, download datasets, train models, deploy models, and repeat.
The Roboflow Python Package is a python wrapper around the core Roboflow web application and REST API.
We also maintain an open source set of CV utililities and notebook tutorials in Python:
- :fire: https://github.com/roboflow/supervision :fire:
- :fire: https://github.com/roboflow/notebooks :fire:
Installation
To install this package, please use Python 3.6 or higher.
Install from PyPi (Recommended):
pip install roboflow
Install from Source:
git clone https://github.com/roboflow-ai/roboflow-python.git
cd roboflow-python
python3 -m venv env
source env/bin/activate
pip3 install -r requirements.txt
Authentication
import roboflow
roboflow.login()
Quickstart
Datasets
Download any of over 200,000 public computer vision datasets from Roboflow Universe. Label and download your own datasets on app.roboflow.com.
import roboflow
dataset = roboflow.download_dataset(dataset_url="universe.roboflow.com/...", model_format="yolov8")
#ex. dataset = roboflow.download_dataset(dataset_url="https://universe.roboflow.com/joseph-nelson/bccd/dataset/1", model_format="yolov8")
print(dataset.location)
Models
Predict with any of over 50,000 public computer vision models. Train your own computer vision models on app.roboflow.com or train upload your model from open source models - see https://github.com/roboflow/notebooks
img_url = "https://media.roboflow.com/quickstart/aerial_drone.jpeg?updatedAt=1678743716455"
universe_model_url = "https://universe.roboflow.com/brad-dwyer/aerial-solar-panels/model/6"
model = roboflow.load_model(model_url=universe_model_url)
pred = model.predict(img_url, hosted=True)
pred.plot()
Library Structure
The Roboflow python library is structured by the core Roboflow application objects.
Workspace (workspace.py) --> Project (project.py) --> Version (version.py)
from roboflow import Roboflow
rf = 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 that 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
If you want to extend our Python library or if you find a bug, please open a PR!
Also be sure to test your code the unittest command at the /root level directory.
Run tests:
python -m unittest
When creating new functions, please follow the Google style Python docstrings. See example below:
def example_function(param1: int, param2: str) -> bool:
"""Example function that does something.
Args:
param1: The first parameter.
param2: The second parameter.
Returns:
The return value. True for success, False otherwise.
"""
We provide a Makefile to format and ensure code quality. Be sure to run them before creating a PR.
# format code with `black` and `isort`
make style
# check code with flake8
make check_code_quality
Note These tests will be run automatically when you commit thanks to git hooks.
4 maintainers
Release files for roboflow 1.1.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| roboflow-1.1.0.tar.gz | 48.8 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| roboflow-1.1.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 105.8 kB
Release files / roboflow-1.1.0.tar.gz
| Download URL | roboflow-1.1.0.tar.gz |
|---|---|
| Size | 48.8 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
7a20bb724262c693c4a9e7977628ab4d5470943e6165846ea7f242e5ced90dc1
|
|
BLAKE2b-256 checksum How to use checksums |
2b356b09a9d5d1aebf9b5e2843c95a586efc29d9c5cb23ef57fe41bac7fe6a76
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/4.0.2 CPython/3.8.17
|
Release files / roboflow-1.1.0-py3-none-any.whl
| Download URL | roboflow-1.1.0-py3-none-any.whl |
|---|---|
| Size | 57.0 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
08d16e595905a0ed9c9e1e0fb8254a18319cdf19990ed03bcde8a419a7ad9b26
|
|
BLAKE2b-256 checksum How to use checksums |
53909a7e6a590003a5bae530559f23069cd18e9a26d9626bc52d94e7056ad895
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/4.0.2 CPython/3.8.17
|
