Awesome TensorFlow
A curated list of awesome TensorFlow experiments, libraries, and projects. Inspired by awesome-machine-learning.
What is TensorFlow?
TensorFlow is an open source software library for numerical computation using data flow graphs. In other words, the best way to build deep learning models.
More info here.
Table of Contents
Tutorials
Models/Projects
-
Tensorflow-Project-Template - A simple and well-designed template for your tensorflow project.
-
Domain Transfer Network - Implementation of Unsupervised Cross-Domain Image Generation
-
Show, Attend and Tell - Attention Based Image Caption Generator
-
Neural Style Implementation of Neural Style
-
SRGAN - Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network
-
Pretty Tensor - Pretty Tensor provides a high level builder API
-
Neural Style - An implementation of neural style
-
AlexNet3D - An implementations of AlexNet3D. Simple AlexNet model but with 3D convolutional layers (conv3d).
-
TensorFlow White Paper Notes - Annotated notes and summaries of the TensorFlow white paper, along with SVG figures and links to documentation
-
NeuralArt - Implementation of A Neural Algorithm of Artistic Style
-
Generative Handwriting Demo using TensorFlow - An attempt to implement the random handwriting generation portion of Alex Graves’ paper
-
Neural Turing Machine in TensorFlow - implementation of Neural Turing Machine
-
GoogleNet Convolutional Neural Network Groups Movie Scenes By Setting - Search, filter, and describe videos based on objects, places, and other things that appear in them
-
Neural machine translation between the writings of Shakespeare and modern English using TensorFlow - This performs a monolingual translation, going from modern English to Shakespeare and vice-versa.
-
Chatbot - Implementation of “A neural conversational model”
-
Seq2seq-Chatbot - Chatbot in 200 lines of code
-
DCGAN - Deep Convolutional Generative Adversarial Networks
-
GAN-CLS -Generative Adversarial Text to Image Synthesis
-
im2im - Unsupervised Image to Image Translation with Generative Adversarial Networks
-
Improved CycleGAN - Unpaired Image to Image Translation
-
DAGAN - Fast Compressed Sensing MRI Reconstruction
-
Colornet - Neural Network to colorize grayscale images - Neural Network to colorize grayscale images
-
Neural Caption Generator - Implementation of “Show and Tell”
-
Neural Caption Generator with Attention - Implementation of “Show, Attend and Tell”
-
Weakly_detector - Implementation of “Learning Deep Features for Discriminative Localization”
-
Dynamic Capacity Networks - Implementation of “Dynamic Capacity Networks”
-
HMM in TensorFlow - Implementation of viterbi and forward/backward algorithms for HMM
-
DeepOSM - Train TensorFlow neural nets with OpenStreetMap features and satellite imagery.
-
DQN-tensorflow - TensorFlow implementation of DeepMind’s ‘Human-Level Control through Deep Reinforcement Learning’ with OpenAI Gym by Devsisters.com
-
Policy Gradient - For Playing Atari Ping Pong
-
Deep Q-Network - For Playing Frozen Lake Game
-
AC - Actor Critic for Playing Discrete Action space Game (Cartpole)
-
A3C - Asynchronous Advantage Actor Critic (A3C) for Continuous Action Space (Bipedal Walker)
-
DAGGER - For Playing Gym Torcs
-
TRPO - For Continuous and Discrete Action Space by
-
Highway Network - TensorFlow implementation of “Training Very Deep Networks” with a blog post
-
Hierarchical Attention Networks - TensorFlow implementation of “Hierarchical Attention Networks for Document Classification”
-
Sentence Classification with CNN - TensorFlow implementation of “Convolutional Neural Networks for Sentence Classification” with a blog post
-
End-To-End Memory Networks - Implementation of End-To-End Memory Networks
-
Character-Aware Neural Language Models - TensorFlow implementation of Character-Aware Neural Language Models
-
YOLO TensorFlow ++ - TensorFlow implementation of ‘YOLO: Real-Time Object Detection’, with training and an actual support for real-time running on mobile devices.
-
Wavenet - This is a TensorFlow implementation of the WaveNet generative neural network architecture for audio generation.
-
Mnemonic Descent Method - Tensorflow implementation of “Mnemonic Descent Method: A recurrent process applied for end-to-end face alignment”
-
CNN visualization using Tensorflow - Tensorflow implementation of “Visualizing and Understanding Convolutional Networks”
-
VGAN Tensorflow - Tensorflow implementation for MIT “Generating Videos with Scene Dynamics” by Vondrick et al.
-
3D Convolutional Neural Networks in TensorFlow - Implementation of “3D Convolutional Neural Networks for Speaker Verification application” in TensorFlow by Torfi et al.
-
U-Net - For Brain Tumor Segmentation
-
Spatial Transformer Networks - Learn the Transformation Function
-
Lip Reading - Cross Audio-Visual Recognition using 3D Architectures in TensorFlow - TensorFlow Implementation of “Cross Audio-Visual Recognition in the Wild Using Deep Learning” by Torfi et al.
-
Attentive Object Tracking - Implementation of “Hierarchical Attentive Recurrent Tracking”
-
Holographic Embeddings for Graph Completion and Link Prediction - Implementation of Holographic Embeddings of Knowledge Graphs
-
Unsupervised Object Counting - Implementation of “Attend, Infer, Repeat”
-
Tensorflow FastText - A simple embedding based text classifier inspired by Facebook’s fastText.
-
MusicGenreClassification - Classify music genre from a 10 second sound stream using a Neural Network.
-
Kubeflow - Framework for easily using Tensorflow with Kubernetes.
-
TensorNets - 40+ Popular Computer Vision Models With Pre-trained Weights.
-
Ladder Network - Implementation of Ladder Network for Semi-Supervised Learning in Keras and Tensorflow
-
TF-Unet - General purpose U-Network implemented in Keras for image segmentation
-
Sarus TF2 Models - A long list of recent generative models implemented in clean, easy to reuse, Tensorflow 2 code (Plain Autoencoder, VAE, VQ-VAE, PixelCNN, Gated PixelCNN, PixelCNN++, PixelSNAIL, Conditional Neural Processes).
-
Model Maker - A transfer learning library that simplifies the process of training, evaluation and deployment for TensorFlow Lite models (support: Image Classification, Object Detection, Text Classification, BERT Question Answer, Audio Classification, Recommendation etc.; API reference).
Powered by TensorFlow
-
YOLO TensorFlow - Implementation of ‘YOLO : Real-Time Object Detection’
-
android-yolo - Real-time object detection on Android using the YOLO network, powered by TensorFlow.
-
Magenta - Research project to advance the state of the art in machine intelligence for music and art generation
Libraries
-
TensorFlow Estimators - high-level TensorFlow API that greatly simplifies machine learning programming (originally tensorflow/skflow )
-
R Interface to TensorFlow - R interface to TensorFlow APIs, including Estimators, Keras, Datasets, etc.
-
Lattice - Implementation of Monotonic Calibrated Interpolated Look-Up Tables in TensorFlow
-
tensorflow.rb - TensorFlow native interface for ruby using SWIG
-
tflearn - Deep learning library featuring a higher-level API
-
TensorLayer - Deep learning and reinforcement learning library for researchers and engineers
-
TensorFlow-Slim - High-level library for defining models
-
TensorFrames - TensorFlow binding for Apache Spark
-
TensorForce - TensorForce: A TensorFlow library for applied reinforcement learning
-
TensorFlowOnSpark - initiative from Yahoo! to enable distributed TensorFlow with Apache Spark.
-
caffe-tensorflow - Convert Caffe models to TensorFlow format
-
keras - Minimal, modular deep learning library for TensorFlow and Theano
-
SyntaxNet: Neural Models of Syntax - A TensorFlow implementation of the models described in Globally Normalized Transition-Based Neural Networks, Andor et al. (2016)
-
keras-js - Run Keras models (tensorflow backend) in the browser, with GPU support
-
NNFlow - Simple framework allowing to read-in ROOT NTuples by converting them to a Numpy array and then use them in Google Tensorflow.
-
Sonnet - Sonnet is DeepMind’s library built on top of TensorFlow for building complex neural networks.
-
tensorpack - Neural Network Toolbox on TensorFlow focusing on training speed and on large datasets.
-
tf-encrypted - Layer on top of TensorFlow for doing machine learning on encrypted data
-
pytorch2keras - Convert PyTorch models to Keras (with TensorFlow backend) format
-
gluon2keras - Convert Gluon models to Keras (with TensorFlow backend) format
-
TensorIO - Lightweight, cross-platform library for deploying TensorFlow Lite models to mobile devices.
-
StellarGraph - Machine Learning on Graphs, a Python library for machine learning on graph-structured (network-structured) data.
-
DeepBay - High-Level Keras Complement for implement common architectures stacks, served as easy to use plug-n-play modules
-
Tensorflow-Probability - Probabalistic programming built on TensorFlow that makes it easy to combine probabilistic models and deep learning on modern hardware.
-
TensorLayerX - TensorLayerX: A Unified Deep Learning Framework for All Hardwares, Backends and OS, including TensorFlow.
-
Speedster - Automatically apply SOTA optimization techniques to achieve the maximum inference speed-up on your hardware.
-
Guild AI - Task runner and package manager for TensorFlow
-
ML Workspace - All-in-one web IDE for machine learning and data science. Combines Tensorflow, Jupyter, VS Code, Tensorboard, and many other tools/libraries into one Docker image.
-
create-tf-app - Project builder command line tool for Tensorflow covering environment management, linting, and logging.
Videos
Papers
Official announcements
Blog posts
Books
-
Machine Learning with TensorFlow by Nishant Shukla, computer vision researcher at UCLA and author of Haskell Data Analysis Cookbook. This book makes the math-heavy topic of ML approachable and practicle to a newcomer.
-
First Contact with TensorFlow by Jordi Torres, professor at UPC Barcelona Tech and a research manager and senior advisor at Barcelona Supercomputing Center
-
Deep Learning with Python - Develop Deep Learning Models on Theano and TensorFlow Using Keras by Jason Brownlee
-
TensorFlow for Machine Intelligence - Complete guide to use TensorFlow from the basics of graph computing, to deep learning models to using it in production environments - Bleeding Edge Press
-
Getting Started with TensorFlow - Get up and running with the latest numerical computing library by Google and dive deeper into your data, by Giancarlo Zaccone
-
Hands-On Machine Learning with Scikit-Learn and TensorFlow – by Aurélien Geron, former lead of the YouTube video classification team. Covers ML fundamentals, training and deploying deep nets across multiple servers and GPUs using TensorFlow, the latest CNN, RNN and Autoencoder architectures, and Reinforcement Learning (Deep Q).
-
Building Machine Learning Projects with Tensorflow – by Rodolfo Bonnin. This book covers various projects in TensorFlow that expose what can be done with TensorFlow in different scenarios. The book provides projects on training models, machine learning, deep learning, and working with various neural networks. Each project is an engaging and insightful exercise that will teach you how to use TensorFlow and show you how layers of data can be explored by working with Tensors.
-
Deep Learning using TensorLayer - by Hao Dong et al. This book covers both deep learning and the implmentation by using TensorFlow and TensorLayer.
-
TensorFlow 2.0 in Action - by Thushan Ganegedara. This practical guide to building deep learning models with the new features of TensorFlow 2.0 is filled with engaging projects, simple language, and coverage of the latest algorithms.
-
Probabilistic Programming and Bayesian Methods for Hackers - by Cameron Davidson-Pilon. Introduction to Bayesian methods and probabalistic graphical models using tensorflow-probability (and, alternatively PyMC2/3).
Contributions
Your contributions are always welcome!
If you want to contribute to this list (please do), send me a pull request or contact me @jtoy
Also, if you notice that any of the above listed repositories should be deprecated, due to any of the following reasons:
- Repository’s owner explicitly say that “this library is not maintained”.
- Not committed for long time (2~3 years).
More info on the guidelines
Credits
- Some of the python libraries were cut-and-pasted from vinta
- The few go reference I found where pulled from this page