import spb.sdk
​
# Print Suite SDK version
print(spb.sdk.__version__)

client = spb.sdk.Client(project_name='<Your Project Name>')
​
# Print project information
print('Project Name: {}'.format(client.get_project_name()))
print('Total number of data: {}'.format(client.get_num_data()))

image_path = '<Local Path to Your Image>'
dataset = '<Your Dataset Name>'
​
client.upload_image(image_path, dataset)
​
# The uploaded image can be seen at the Suite Web UI

bucket = '<Your S3 Bucket Name>'
image_path = '<Path to your Image in the Bucket>'
dataset = '<Your Dataset Name>'
​
client.upload_image_s3(bucket, image_path, dataset)
​
# The uploaded image can be seen at the Suite Web UI

# Generate iterator from selected project
data_iter = client.get_data_page(page_idx='<Index of pages>', page_size='<Number of data in page>')
data = next(data_iter)
​
# Print data information
print('Key: {}'.format(data.get_key()))
print('Dataset: {}'.format(data.get_dataset_name()))
print('Image URL: {}'.format(data.get_image_url()))
print('Status: {}'.format(data.get_status()))

image = data.get_image()
​
# Print image information
print('Image type: {}'.format(image.__class__))
print('Image dtype: {}'.format(image.dtype))
print('Image shape: {}'.format(image.shape))
​
# Plot the image
import matplotlib.pyplot as plt
plt.imshow(image)

download_path, result = data.download_image(download_to='./test.jpg')
​
# Print the HTTP response
print(result)
​
# Plot the downloaded image
import PIL.Image
image = PIL.Image.open(download_path)
plt.imshow(image)

dataset = 'MSCOCO_dataset_a'
​
# Total number of data in your dataset
num_data = client.get_num_data(dataset=dataset)
print('Size of "{}" dataset: {}'.format(dataset, num_data))
​
# Get data iterator in the dataset
data_iter = client.get_data_page(page_idx='<Index of pages>', page_size='<Number of data in page>', dataset=dataset)
data = next(data_iter)

tags = ['<Label Tag 1>', '<Label Tag 2>', ...]
​
# Number of data containing all the tags
num_data = client.get_num_data(tags=tags)
print('Number of data containing tags {}: {}'.format(tags, num_data))
​
# Get data iterator in the filtered list of data
data_iter = client.get_data_page(page_idx='<Index of pages>', page_size='<Number of data in page>', tags=tags)
data = next(data_iter)
​
# Print tags of the data
print('Tags: {}'.format(data.get_tags()))

category_labels = data.get_category_labels()
object_labels = data.get_object_labels()
​
# Print labels
print('category_labels', category_labels)
print('object_labels', object_labels)
# Example output:
#   category_labels ['bugatti']
#   object_labels [
#       {'id': 1, 'class': 'bugatti',
#        'shape': {'box': {'x': 83, 'y': 52, 'width': 819, 'height': 457}},
#        'properties': []},
#       {'id': 2, 'class': ...},
#       {'id': 3, 'class': ...},
#       ...
#   ]

category_labels = ['teslaroadster']
object_labels[0]['class'] = 'teslaroadster'
​
# Print labels
print('category_labels', category_labels)
print('object_labels', object_labels)
# Example output:
#   category_labels ['teslaroadster']
#   object_labels [
#       {'id': 1, 'class': 'teslaroadster',
#        'shape': {'box': {'x': 83, 'y': 52, 'width': 819, 'height': 457}},
#        'properties': []},
#       {'id': 2, 'class': ...},
#       {'id': 3, 'class': ...},
#       ...
#   ]

data.set_category_labels(category_labels)
data.set_object_labels(object_labels)
​
# The change can be seen at the Suite Web UI

# Get the data row from the Suite
data = client.get_data(data_idx=0)
​
# Print labels
print('category_labels', data.get_category_labels())
print('object_labels', data.get_object_labels())
# Example output:
#   Same with the labels at local