TensorBay Python SDK

TensorBay Python SDK is a python library to access TensorBay and manage your datasets.
It provides:

• A pythonic way to access your TensorBay resources by TensorBay OpenAPI.
• An easy-to-use CLI tool gas (Graviti AI service) to communicate with TensorBay.
• A consistent dataset format to read and write your datasets.

Installation

pip3 install tensorbay

Documentation

More information can be found on the documentation site

Usage

An AccessKey is needed to communicate with TensorBay. Please visit this page to get an AccessKey first.

Authorize a client object

fromtensorbayimportGASgas=GAS("<YOUR_ACCESSKEY>")

Create a Dataset

gas.create_dataset("DatasetName")

List Dataset names

dataset_names=gas.list_dataset_names()

Upload images to the Dataset

fromtensorbay.datasetimportData, Dataset# Organize the local dataset by the "Dataset" class before uploading.dataset=Dataset("DatasetName")
# TensorBay uses "segment" to separate different parts in a dataset.segment=dataset.create_segment()
segment.append(Data("0000001.jpg"))
segment.append(Data("0000002.jpg"))
dataset_client=gas.upload_dataset(dataset)
# TensorBay provides dataset version control feature, commit the uploaded data before using it.dataset_client.commit("Initial commit")

Read images from the Dataset

fromPILimportImagefromtensorbay.datasetimportSegmentdataset_client=gas.get_dataset("DatasetName")
segment=Segment("", dataset_client)
fordatainsegment:
withdata.open() asfp:
image=Image.open(fp)
width, height=image.sizeimage.show()

Delete the Dataset

gas.delete_dataset("DatasetName")