roboflow 0.0.2

Loader for Roboflow datasets.

Roboflow Python Library

This is a helper library to load your Roboflow datasets into your python scripts and Jupyter notebooks.

Requirements

This package requires python >=3.6 and a (free) Roboflow account.

Installing

With PIP

pip install roboflow

With Anaconda

conda install roboflow

Setup

The roboflow package works in conjunction with your Roboflow account.

From your Account page, click Roboflow Keys to get your API key. You can then use the roboflow python package to manage downloading your datasets in various formats.

import roboflow
roboflow.auth("<<YOUR API KEY>>")
info = roboflow.load("chess-sample", 1, "tfrecord")

# dataset is now downloaded and unzipped in your current directory
# and info contains the paths you need to load it into your favorite
# machine learning libraries

By default the folder is named

${dataset-name}.${version-number}-${version-name}.${format}

(For example, Chess Sample.v1-small-gray.coco).

The file hierarchy is three folders containing the train, valid, and test data you selected in the Roboflow upload flow (and the format you specified in roboflow.load above). There is also a README.roboflow.txt describing the preprocessing and augmentation steps and, optionally, a README.dataset.txt provided by the person who shared the dataset.

Doing Inference

It's important to pre-process your images for inference the same way you pre-processed your training images. For this, get a pre-processor via the roboflow.infer method which will return a function you can use to pre-process your images.

import roboflow
roboflow.auth("<<YOUR API KEY>>")
process = roboflow.preprocessor("chess-sample", 1)
images = process.fromFile("example.jpg") # returns a numpy array (of 1 image, unless you used tiling)

Benefits

This package currently provides two main benefits over downloading and loading your datasets manually.

1. If you have previously loaded your dataset, it will automatically use the local copy rather than re-downloading.
2. You can dynamically choose the export format at runtime rather than export-time.

Roadmap

We plan to include more features in the future to allow you (for example, to let you easily do inference on your trained models).