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.
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
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
- Paper: Introduced by Danielczuk et al. in Segmenting Unknown 3D Objects from Real Depth Images using Mask R-CNN Trained on Synthetic Data
- Point of Contact: N/A
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.
Last modified 27d ago
Copy link
Contents
WISDOM dataset
What is WISDOM Dataset?
Download WISDOM Dataset in Python
Load WISDOM Dataset Training Subset in Python
Load WISDOM Dataset Testing Subset in Python
WISDOM Dataset Structure
How to use WISDOM Dataset with PyTorch and TensorFlow in Python
Additional Information about WISDOM Dataset
WISDOM Dataset Description
WISDOM Dataset FAQs