37

GitHub - tensorflow/datasets: A collection of datasets ready to use with TensorF...

 5 years ago
source link: https://github.com/tensorflow/datasets
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.

README.md

TensorFlow Datasets

Note: tensorflow_datasets is not yet functional and is under active development. API unstable.

TensorFlow Datasets provides many public datasets as tf.data.Datasets.

Travis

Installation:

pip install tensorflow-datasets
# Also requires tensorflow or tensorflow-gpu to be installed

Usage:

import tensorflow_datasets as datasets

# Construct a tf.data.Dataset
dataset = datasets.load("mnist",
                        split="train",
                        data_dir="~/tfdata",
                        download=True)

# Build your input pipeline
dataset = dataset.shuffle(1000).batch(128).prefetch(1)
features = dataset.make_oneshot_iterator().get_next()
image, label = features["input"], features["output"]

report

Recommend

About Joyk


Aggregate valuable and interesting links.
Joyk means Joy of geeK