hub 0.9.0.3

Snark Hub

The fastest way to access and manage datasets for PyTorch and TensorFlow

Hub provides fast access to the state-of-the-art datasets for Deep Learning, enabling data scientists to manage them, build scalable data pipelines and connect to Pytorch and Tensorflow

Problems with Current Workflows

We realized that there are a few problems related with current workflow in deep learning data management through our experience of working with deep learning companies and researchers. Most of the time Data Scientists/ML researchers work on data management and preprocessing instead of doing modeling. Deep Learning often requires to work with large datasets. Those datasets can grow up to terabyte or even petabyte size. It is hard to manage data, version control and track. It is time-consuming to download the data and link with the training or inference code. There is no easy way to access a chunk of it and possibly visualize. Wouldn’t it be more convenient to have large datasets stored & version-controlled as single numpy-like array on the cloud and have access to it from any machine at scale?

Getting Started

Access public data. Fast

We’ve talked the talk, now let’s walk through how it works:

pip3 install hub

You can access public datasets with a few lines of code.

import hub

mnist = hub.load("mnist/mnist")
mnist["data"][0:1000].compute()

Train a model

Load the data and directly train your model using pytorch

import hub
import pytorch

cifar = hub.load("cifar/cifar10")
cifar = cifar.to_pytorch()

train_loader = torch.utils.data.DataLoader(
        cifar, batch_size=1, num_workers=0, collate_fn=cifar.collate_fn
)

for images, labels in train_loader:
    # your training loop here

Upload your dataset and access it from anywhere in 3 simple steps

1. Register a free account at Snark and authenticate locally

hub login

2. Then create a dataset and upload

from hub import tensor, dataset

images = tensor.from_array(np.zeros((4, 512, 512)))
labels = tensor.from_array(np.zeros((4, 512, 512)))

ds = dataset.from_tensors({"images": images, "labels": labels})
ds.store("username/basic")

3. Access it from anywhere else in the world, on any device having a command line.

import hub

ds = hub.load("username/basic")

For more advanced data pipelines like uploading large datasets or applying many transformations, please see docs.

Things you can do with Hub

• Store large datasets with version control
• Collaborate as in Google Docs: Multiple data scientists working on the same data in sync with no interruptions
• Access from multiple machines simultaneously
• Integration with your ML tools like Numpy, Dask, PyTorch, or TensorFlow.
• Create arrays as big as you want
• Take a quick look on your data without redundant manipulations/in a matter of seconds/etc.

Use Cases

• Aerial images: Satellite and drone imagery
• Medical Images: Volumetric images such as MRI or Xray
• Self-Driving Cars: Radar, 3D LIDAR, Point Cloud, Semantic Segmentation, Video Objects
• Retail: Self-checkout datasets
• Media: Images, Video, Audio storage

Examples

Snark’s Hub format lets you achieve faster inference at a lower cost. Test out the datasets we’ve converted into Hub format - see for yourself!

• Waymo Open Dataset
• Aptiv nuScenes