Neural Network Toolbox on TensorFlow
Project description
.. figure:: .github/tensorpack.png
:alt: Tensorpack
Tensorpack
Tensorpack is a training interface based on TensorFlow.
|Build Status| |ReadTheDoc| |Gitter chat| |model-zoo|
Features:
---------
It's Yet Another TF high-level API, with **speed**, **readability** and
**flexibility** built together.
1. Focus on **training speed**.
- Speed comes for free with tensorpack -- it uses TensorFlow in the
**efficient way** with no extra overhead. On different CNNs, it
runs training `1.2~5x
faster <https://github.com/tensorpack/benchmarks/tree/master/other-wrappers>`__
than the equivalent Keras code.
- Data-parallel multi-GPU/distributed training strategy is
off-the-shelf to use. It scales as well as Google's `official
benchmark <https://www.tensorflow.org/performance/benchmarks>`__.
- See
`tensorpack/benchmarks <https://github.com/tensorpack/benchmarks>`__
for some benchmark scripts.
2. Focus on **large datasets**.
- `You don't usually need
``tf.data`` <http://tensorpack.readthedocs.io/tutorial/input-source.html#tensorflow-reader-cons>`__.
Symbolic programming often makes data processing harder.
Tensorpack helps you efficiently process large datasets (e.g.
ImageNet) in **pure Python** with autoparallelization.
3. It's not a model wrapper.
- There are too many symbolic function wrappers in the world.
Tensorpack includes only a few common models. But you can use any
symbolic function library inside tensorpack, including
tf.layers/Keras/slim/tflearn/tensorlayer/....
See
`tutorials <http://tensorpack.readthedocs.io/tutorial/index.html#user-tutorials>`__
to know more about these features.
`Examples <examples>`__:
------------------------
We refuse toy examples. Instead of showing you 10 arbitrary networks
trained on toy datasets, `tensorpack examples <examples>`__ faithfully
replicate papers and care about reproducing numbers, demonstrating its
flexibility for actual research.
Vision:
~~~~~~~
- `Train ResNet <examples/ResNet>`__ and `other
models <examples/ImageNetModels>`__ on ImageNet.
- `Train Faster-RCNN / Mask-RCNN on COCO object
detection <examples/FasterRCNN>`__
- `Generative Adversarial Network(GAN) variants <examples/GAN>`__,
including DCGAN, InfoGAN, Conditional GAN, WGAN, BEGAN, DiscoGAN,
Image to Image, CycleGAN.
- `DoReFa-Net: train binary / low-bitwidth CNN on
ImageNet <examples/DoReFa-Net>`__
- `Fully-convolutional Network for Holistically-Nested Edge
Detection(HED) <examples/HED>`__
- `Spatial Transformer Networks on MNIST
addition <examples/SpatialTransformer>`__
- `Visualize CNN saliency maps <examples/Saliency>`__
- `Similarity learning on MNIST <examples/SimilarityLearning>`__
Reinforcement Learning:
~~~~~~~~~~~~~~~~~~~~~~~
- `Deep Q-Network(DQN) variants on Atari
games <examples/DeepQNetwork>`__, including DQN, DoubleDQN,
DuelingDQN.
- `Asynchronous Advantage Actor-Critic(A3C) with demos on OpenAI
Gym <examples/A3C-Gym>`__
Speech / NLP:
~~~~~~~~~~~~~
- `LSTM-CTC for speech recognition <examples/CTC-TIMIT>`__
- `char-rnn for fun <examples/Char-RNN>`__
- `LSTM language model on PennTreebank <examples/PennTreebank>`__
Install:
--------
Dependencies:
- Python 2.7 or 3
- Python bindings for OpenCV (Optional, but required by a lot of
features)
- TensorFlow >= 1.3.0 (Optional if you only want to use
``tensorpack.dataflow`` alone as a data processing library)
::
# install git, then:
pip install -U git+https://github.com/tensorpack/tensorpack.git
# or add `--user` to avoid system-wide installation.
Citing Tensorpack:
------------------
If you use Tensorpack in your research or wish to refer to the examples,
please cite with:
::
@misc{wu2016tensorpack,
title={Tensorpack},
author={Wu, Yuxin and others},
howpublished={\url{https://github.com/tensorpack/}},
year={2016}
}
.. |Build Status| image:: https://travis-ci.org/tensorpack/tensorpack.svg?branch=master
:target: https://travis-ci.org/tensorpack/tensorpack
.. |ReadTheDoc| image:: https://readthedocs.org/projects/tensorpack/badge/?version=latest
:target: http://tensorpack.readthedocs.io/en/latest/index.html
.. |Gitter chat| image:: https://badges.gitter.im/gitterHQ/gitter.png
:target: https://gitter.im/tensorpack/users
.. |model-zoo| image:: https://img.shields.io/badge/model-zoo-brightgreen.svg
:target: http://models.tensorpack.com
:alt: Tensorpack
Tensorpack
Tensorpack is a training interface based on TensorFlow.
|Build Status| |ReadTheDoc| |Gitter chat| |model-zoo|
Features:
---------
It's Yet Another TF high-level API, with **speed**, **readability** and
**flexibility** built together.
1. Focus on **training speed**.
- Speed comes for free with tensorpack -- it uses TensorFlow in the
**efficient way** with no extra overhead. On different CNNs, it
runs training `1.2~5x
faster <https://github.com/tensorpack/benchmarks/tree/master/other-wrappers>`__
than the equivalent Keras code.
- Data-parallel multi-GPU/distributed training strategy is
off-the-shelf to use. It scales as well as Google's `official
benchmark <https://www.tensorflow.org/performance/benchmarks>`__.
- See
`tensorpack/benchmarks <https://github.com/tensorpack/benchmarks>`__
for some benchmark scripts.
2. Focus on **large datasets**.
- `You don't usually need
``tf.data`` <http://tensorpack.readthedocs.io/tutorial/input-source.html#tensorflow-reader-cons>`__.
Symbolic programming often makes data processing harder.
Tensorpack helps you efficiently process large datasets (e.g.
ImageNet) in **pure Python** with autoparallelization.
3. It's not a model wrapper.
- There are too many symbolic function wrappers in the world.
Tensorpack includes only a few common models. But you can use any
symbolic function library inside tensorpack, including
tf.layers/Keras/slim/tflearn/tensorlayer/....
See
`tutorials <http://tensorpack.readthedocs.io/tutorial/index.html#user-tutorials>`__
to know more about these features.
`Examples <examples>`__:
------------------------
We refuse toy examples. Instead of showing you 10 arbitrary networks
trained on toy datasets, `tensorpack examples <examples>`__ faithfully
replicate papers and care about reproducing numbers, demonstrating its
flexibility for actual research.
Vision:
~~~~~~~
- `Train ResNet <examples/ResNet>`__ and `other
models <examples/ImageNetModels>`__ on ImageNet.
- `Train Faster-RCNN / Mask-RCNN on COCO object
detection <examples/FasterRCNN>`__
- `Generative Adversarial Network(GAN) variants <examples/GAN>`__,
including DCGAN, InfoGAN, Conditional GAN, WGAN, BEGAN, DiscoGAN,
Image to Image, CycleGAN.
- `DoReFa-Net: train binary / low-bitwidth CNN on
ImageNet <examples/DoReFa-Net>`__
- `Fully-convolutional Network for Holistically-Nested Edge
Detection(HED) <examples/HED>`__
- `Spatial Transformer Networks on MNIST
addition <examples/SpatialTransformer>`__
- `Visualize CNN saliency maps <examples/Saliency>`__
- `Similarity learning on MNIST <examples/SimilarityLearning>`__
Reinforcement Learning:
~~~~~~~~~~~~~~~~~~~~~~~
- `Deep Q-Network(DQN) variants on Atari
games <examples/DeepQNetwork>`__, including DQN, DoubleDQN,
DuelingDQN.
- `Asynchronous Advantage Actor-Critic(A3C) with demos on OpenAI
Gym <examples/A3C-Gym>`__
Speech / NLP:
~~~~~~~~~~~~~
- `LSTM-CTC for speech recognition <examples/CTC-TIMIT>`__
- `char-rnn for fun <examples/Char-RNN>`__
- `LSTM language model on PennTreebank <examples/PennTreebank>`__
Install:
--------
Dependencies:
- Python 2.7 or 3
- Python bindings for OpenCV (Optional, but required by a lot of
features)
- TensorFlow >= 1.3.0 (Optional if you only want to use
``tensorpack.dataflow`` alone as a data processing library)
::
# install git, then:
pip install -U git+https://github.com/tensorpack/tensorpack.git
# or add `--user` to avoid system-wide installation.
Citing Tensorpack:
------------------
If you use Tensorpack in your research or wish to refer to the examples,
please cite with:
::
@misc{wu2016tensorpack,
title={Tensorpack},
author={Wu, Yuxin and others},
howpublished={\url{https://github.com/tensorpack/}},
year={2016}
}
.. |Build Status| image:: https://travis-ci.org/tensorpack/tensorpack.svg?branch=master
:target: https://travis-ci.org/tensorpack/tensorpack
.. |ReadTheDoc| image:: https://readthedocs.org/projects/tensorpack/badge/?version=latest
:target: http://tensorpack.readthedocs.io/en/latest/index.html
.. |Gitter chat| image:: https://badges.gitter.im/gitterHQ/gitter.png
:target: https://gitter.im/tensorpack/users
.. |model-zoo| image:: https://img.shields.io/badge/model-zoo-brightgreen.svg
:target: http://models.tensorpack.com
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
tensorpack-0.8.6.tar.gz
(185.3 kB
view hashes)
Built Distribution
tensorpack-0.8.6-py2.py3-none-any.whl
(253.0 kB
view hashes)
Close
Hashes for tensorpack-0.8.6-py2.py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4e0dc643cf283e959f8539554dd7af53dc21d04b0b38c487824e1cf5a69e7eca |
|
MD5 | b7df9c31bbfc2b4fc4003f4f543fc79e |
|
BLAKE2b-256 | 545e5f4c78a7becda59d8cd1b82cff9d8819a0057f17fceba237b2bbfdddc99d |