v2.5.0
Datasets ⭐
EXAMPLE CODE
WISDOM Dataset
Load WISDOM dataset in Python fast with one line of code. 400 images dataset with masks. Stream WISDOM Dataset while training models in Pytorch & Tensorflow.
Visualization of WISDOM dataset on the Activeloop Platform

WISDOM dataset

What is WISDOM Dataset?

The WISDOM (Warehouse Instance Segmentation Dataset for Object Manipulation) dataset is a synthetic training dataset of 50,000 depth images and 320,000 object masks using simulated heaps of 3D CAD models.

Download WISDOM Dataset in Python

Instead of downloading the WISDOM dataset in Python, you can effortlessly load it in Python via our open-source package Hub with just one line of code.

Load WISDOM Dataset Training Subset in Python

1
import hub
2
ds = hub.load('hub://activeloop/wisdom-real-highres-train')
Copied!

Load WISDOM Dataset Testing Subset in Python

1
import hub
2
ds = hub.load('hub://activeloop/wisdom-real-highres-test')
Copied!

WISDOM Dataset Structure

WISDOM Data Fields

  • images: tensor containing the image.
  • masks: tensor to represent image masks.

WISDOM Data Splits

How to use WISDOM Dataset with PyTorch and TensorFlow in Python

Train a model on WISDOM dataset with PyTorch in Python

Let's use Hub's built-in PyTorch one-line dataloader to connect the data to the compute:
1
dataloader = ds.pytorch(num_workers=0, batch_size=4, shuffle=False)
Copied!

Train a model on WISDOM dataset with TensorFlow in Python

1
dataloader = ds.tensorflow()
Copied!

Additional Information about WISDOM Dataset

WISDOM Dataset Description

WISDOM Dataset Curators

Michael Danielczuk, Matthew Matl, Saurabh Gupta, Andrew Li, Andrew Lee, Jeffrey Mahler, Ken Goldberg

WISDOM Dataset Licensing Information

Hub users may have access to a variety of publicly available datasets. We do not host or distribute these datasets, vouch for their quality or fairness, or claim that you have a license to use the datasets. It is your responsibility to determine whether you have permission to use the datasets under their license.
If you're a dataset owner and do not want your dataset to be included in this library, please get in touch through a GitHub issue. Thank you for your contribution to the ML community!

WISDOM Dataset Citation Information

1
@article{DBLP:journals/corr/abs-1809-05825,
2
author = {Michael Danielczuk and
3
Matthew Matl and
4
Saurabh Gupta and
5
Andrew Li and
6
Andrew Lee and
7
Jeffrey Mahler and
8
Ken Goldberg},
9
title = {Segmenting Unknown 3D Objects from Real Depth Images using Mask {R-CNN}
10
Trained on Synthetic Point Clouds},
11
journal = {CoRR},
12
volume = {abs/1809.05825},
13
year = {2018},
14
url = {http://arxiv.org/abs/1809.05825},
15
eprinttype = {arXiv},
16
eprint = {1809.05825},
17
timestamp = {Sat, 23 Jan 2021 01:11:29 +0100},
18
biburl = {https://dblp.org/rec/journals/corr/abs-1809-05825.bib},
19
bibsource = {dblp computer science bibliography, https://dblp.org}
20
}
Copied!

WISDOM Dataset FAQs

What is the WISDOM dataset for Python?

The WISDOM dataset is a synthetic training dataset of 50,000 depth images. The dataset also has 320,000 object masks using simulated heaps of 3D CAD models.

How can I use the WISDOM dataset in PyTorch or TensorFlow?

You can stream the WISDOM dataset while training a model in PyTorch or TensorFlow with one line of code using the open-source package Activeloop Hub in Python. See detailed instructions on how to train a model on WISDOM dataset with PyTorch in Python or train a model on WISDOM dataset with TensorFlow in Python.