Skip to content
Branch: master
Find file History
auslaner and dynamicwebpaige Fixed #6183; Typo Correction and Pyglet Docs link (#6328)
Added missing 's' character in "install". Added link to pyglet documentation.
Latest commit 1aa241d Mar 26, 2019
Type Name Latest commit message Commit time
Failed to load latest commit information.
a3c_blogpost Fixed #6183; Typo Correction and Pyglet Docs link (#6328) Mar 26, 2019
adv_imagenet_models Update in adv_imagenet_models. Dec 19, 2017
adversarial_crypto A few more formatting fixes. Oct 27, 2017
adversarial_logit_pairing Adversarial logit pairing - update table formatting Jul 10, 2018
adversarial_text from six.moves import xrange (en masse) AGAIN Mar 5, 2018
astronet Move tensorflow_models/research/astronet to google-research/exoplanet-ml Dec 4, 2018
audioset Shortened information about used library versions in Audioset README Nov 12, 2018
autoaugment Added autoaugment model. Sep 19, 2018
autoencoder Bugfix for initialization in autoencoder. (#3477) Apr 17, 2018
brain_coder long was removed in Python 3 (en masse) Jan 22, 2018
cognitive_mapping_and_planning Typo: resnet_v2_50_path --> rgb_resnet_v2_50_path (#3273) Feb 24, 2018
cognitive_planning adding implementation of (#5265) Sep 11, 2018
compression MS-SSIM: Read image data as binary Mar 24, 2018
cvt_text Fix cvt_text citation. Oct 2, 2018
deep_contextual_bandits Readme changes. Jul 23, 2018
deep_speech Add script for deep speech benchmark (#4750) Jul 12, 2018
deeplab Fixed improper HNASNet architecture (#6419) Mar 22, 2019
delf Merge pull request #3918 from andrefaraujo/master Apr 9, 2018
differential_privacy remove all code related to differential privacy (#6045) Jan 15, 2019
domain_adaptation from six.moves import xrange (en masse) Jan 20, 2018
efficient-hrl add training code Dec 5, 2018
feelvos Update the codes due to recent change in DeepLab. (#6287) Feb 28, 2019
fivo Whitespace in license for Sep 7, 2018
gan Merge pull request #5401 from joel-shor/master Oct 2, 2018
global_objectives Added new model, global objectives. Jun 13, 2018
im2txt speedup caption generator on im2txt (#6255) Mar 6, 2019
inception Added missing python interpreter directive (#6015) Jan 11, 2019
keypointnet Adding keypointnet to research/models. Jul 24, 2018
learned_optimizer from six.moves import xrange (en masse) Jan 20, 2018
learning_to_remember_rare_events Changes to "Learning to Remember..." to make it runnable in Python3.5 Mar 7, 2018
learning_unsupervised_learning Modify flag name for the checkpoint path Aug 14, 2018
lexnet_nc Add link to the paper and the Tratz dataset (#3745) Apr 12, 2018
lfads [LFADS] Fixes typo in (#6161) Feb 6, 2019
lm_1b Remove contacts of exGoogler Jan 31, 2018
lm_commonsense Add link to arxiv paper. Aug 3, 2018
lstm_object_detection Code clean up. Fix typo. Dec 3, 2018
marco Adding panoptic evaluation tools and update internal changes. (#6320) Mar 7, 2019
maskgan Increase minimum TF version for DEFINE_enum and rename variable mappi… Apr 30, 2018
minigo Add run individual step only option (#4049) May 4, 2018
morph_net Redirect MorphNet project to the new repo. (#6418) Mar 21, 2019
namignizer Move the research models into a research subfolder (#2430) Sep 21, 2017
neural_gpu resolve 'float' to 'int' conversion, StringIO import, conacat str + byte May 22, 2018
neural_programmer Fix Python 3 Syntax Errors (en masse) Jan 24, 2018
next_frame_prediction Remove contacts of exGoogler Jan 31, 2018
nst_blogpost Update 4_Neural_Style_Transfer_with_Eager_Execution.ipynb Nov 5, 2018
object_detection Merged commit includes the following changes: (#6315) Mar 7, 2019
pcl_rl fix bugs in Python 3.X Mar 22, 2018
ptn from six.moves import xrange (en masse) Jan 20, 2018
qa_kg Move the research models into a research subfolder (#2430) Sep 21, 2017
rebar Move the research models into a research subfolder (#2430) Sep 21, 2017
resnet Yet another ResNet (#4298) May 17, 2018
seq2species Rename programmers_guide/ to guide/ in tf-models. Jun 26, 2018
skip_thoughts Prevents TypeError:'dict_keys' object does not support indexing Oct 11, 2018
slim Merged commit includes the following changes: (#6407) Mar 21, 2019
steve Fixed minor bug in STEVE which sometimes prevented resuming experiments. Jul 18, 2018
street Fix Python 3 Syntax Errors (en masse) Jan 23, 2018
struct2depth Added struct2depth model Nov 19, 2018
swivel Fixed links to evaluation data in makefile (#5402) Oct 5, 2018
syntaxnet Syntaxnet/DRAGNN visualization: Update webpack and (necessarily) babe… Jan 16, 2019
tcn Rename programmers_guide/ to guide/ in tf-models. Jun 26, 2018
tensorrt Update (#6034) Jan 14, 2019
textsum Remove contacts of exGoogler Jan 31, 2018
transformer Restore blank line Nov 21, 2017
vid2depth Fix README links. Jun 5, 2018
video_prediction Move the research models into a research subfolder (#2430) Sep 21, 2017 Added struct2depth model Nov 19, 2018 Add required packages which are missing. (#4087) Apr 25, 2018

TensorFlow Research Models

This folder contains machine learning models implemented by researchers in TensorFlow. The models are maintained by their respective authors. To propose a model for inclusion, please submit a pull request.


  • adversarial_crypto: protecting communications with adversarial neural cryptography.
  • adversarial_text: semi-supervised sequence learning with adversarial training.
  • attention_ocr: a model for real-world image text extraction.
  • audioset: Models and supporting code for use with AudioSet.
  • autoencoder: various autoencoders.
  • brain_coder: Program synthesis with reinforcement learning.
  • cognitive_mapping_and_planning: implementation of a spatial memory based mapping and planning architecture for visual navigation.
  • compression: compressing and decompressing images using a pre-trained Residual GRU network.
  • cvt_text: semi-supervised sequence learning with cross-view training.
  • deep_contextual_bandits: code for a variety of contextual bandits algorithms using deep neural networks and Thompson sampling.
  • deep_speech: automatic speech recognition.
  • deeplab: deep labeling for semantic image segmentation.
  • delf: deep local features for image matching and retrieval.
  • differential_privacy: differential privacy for training data.
  • domain_adaptation: domain separation networks.
  • fivo: filtering variational objectives for training generative sequence models.
  • gan: generative adversarial networks.
  • im2txt: image-to-text neural network for image captioning.
  • inception: deep convolutional networks for computer vision.
  • keypointnet: discovery of latent 3D keypoints via end-to-end geometric eeasoning [demo].
  • learning_to_remember_rare_events: a large-scale life-long memory module for use in deep learning.
  • learning_unsupervised_learning: a meta-learned unsupervised learning update rule.
  • lexnet_nc: a distributed model for noun compound relationship classification.
  • lfads: sequential variational autoencoder for analyzing neuroscience data.
  • lm_1b: language modeling on the one billion word benchmark.
  • lm_commonsense: commonsense reasoning using language models.
  • maskgan: text generation with GANs.
  • namignizer: recognize and generate names.
  • neural_gpu: highly parallel neural computer.
  • neural_programmer: neural network augmented with logic and mathematic operations.
  • next_frame_prediction: probabilistic future frame synthesis via cross convolutional networks.
  • object_detection: localizing and identifying multiple objects in a single image.
  • pcl_rl: code for several reinforcement learning algorithms, including Path Consistency Learning.
  • ptn: perspective transformer nets for 3D object reconstruction.
  • marco: automating the evaluation of crystallization experiments.
  • qa_kg: module networks for question answering on knowledge graphs.
  • real_nvp: density estimation using real-valued non-volume preserving (real NVP) transformations.
  • rebar: low-variance, unbiased gradient estimates for discrete latent variable models.
  • resnet: deep and wide residual networks.
  • seq2species: deep learning solution for read-level taxonomic classification.
  • skip_thoughts: recurrent neural network sentence-to-vector encoder.
  • slim: image classification models in TF-Slim.
  • street: identify the name of a street (in France) from an image using a Deep RNN.
  • struct2depth: unsupervised learning of depth and ego-motion.
  • swivel: the Swivel algorithm for generating word embeddings.
  • syntaxnet: neural models of natural language syntax.
  • tcn: Self-supervised representation learning from multi-view video.
  • textsum: sequence-to-sequence with attention model for text summarization.
  • transformer: spatial transformer network, which allows the spatial manipulation of data within the network.
  • vid2depth: learning depth and ego-motion unsupervised from raw monocular video.
  • video_prediction: predicting future video frames with neural advection.
You can’t perform that action at this time.