Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit ee6901f

Browse files
author
Barret Zoph
authored
Merge branch 'master' into master
2 parents 5a66a39 + 8ffcc2f commit ee6901f

File tree

1,921 files changed

+264871
-15558
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

1,921 files changed

+264871
-15558
lines changed

.gitignore

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -90,3 +90,8 @@ ENV/
9090

9191
# PyCharm
9292
.idea/
93+
94+
# For mac
95+
.DS_Store
96+
97+
samples/outreach/blogs/segmentation_blogpost/carvana-image-masking-challenge/

CODEOWNERS

Lines changed: 64 additions & 40 deletions
Original file line numberDiff line numberDiff line change
@@ -1,40 +1,64 @@
1-
official/* @nealwu @k-w-w @jhseu @itsmeolivia
2-
research/adversarial_crypto/* @dave-andersen
3-
research/adversarial_text/* @rsepassi
4-
research/adv_imagenet_models/* @AlexeyKurakin
5-
research/attention_ocr/* @alexgorban
6-
research/audioset/* @plakal @dpwe
7-
research/autoaugment/* @barretzoph
8-
research/autoencoders/* @snurkabill
9-
research/cognitive_mapping_and_planning/* @s-gupta
10-
research/compression/* @nmjohn
11-
research/delf/* @andrefaraujo
12-
research/differential_privacy/* @panyx0718
13-
research/domain_adaptation/* @bousmalis @ddohan
14-
research/im2txt/* @cshallue
15-
research/inception/* @shlens @vincentvanhoucke
16-
research/learned_optimizer/* @olganw @nirum
17-
research/learning_to_remember_rare_events/* @lukaszkaiser @ofirnachum
18-
research/lfads/* @jazcollins @susillo
19-
research/lm_1b/* @oriolvinyals @panyx0718
20-
research/namignizer/* @knathanieltucker
21-
research/neural_gpu/* @lukaszkaiser
22-
research/neural_programmer/* @arvind2505
23-
research/next_frame_prediction/* @panyx0718
24-
research/object_detection/* @jch1 @tombstone @derekjchow @jesu9 @dreamdragon
25-
research/pcl_rl/* @ofirnachum
26-
research/ptn/* @xcyan @arkanath @hellojas @honglaklee
27-
research/real_nvp/* @laurent-dinh
28-
research/rebar/* @gjtucker
29-
research/resnet/* @panyx0718
30-
research/skip_thoughts/* @cshallue
31-
research/slim/* @sguada @nathansilberman
32-
research/street/* @theraysmith
33-
research/swivel/* @waterson
34-
research/syntaxnet/* @calberti @andorardo @bogatyy @markomernick
35-
research/textsum/* @panyx0718 @peterjliu
36-
research/transformer/* @daviddao
37-
research/video_prediction/* @cbfinn
38-
tutorials/embedding/* @zffchen78 @a-dai
39-
tutorials/image/* @sherrym @shlens
40-
tutorials/rnn/* @lukaszkaiser @ebrevdo
1+
* @tensorflow/tf-garden-team
2+
/official/ @tensorflow/tf-garden-team @karmel
3+
/research/adversarial_crypto/ @dave-andersen
4+
/research/adversarial_logit_pairing/ @AlexeyKurakin
5+
/research/adversarial_text/ @rsepassi @a-dai
6+
/research/adv_imagenet_models/ @AlexeyKurakin
7+
/research/attention_ocr/ @alexgorban
8+
/research/audioset/ @plakal @dpwe
9+
/research/autoaugment/* @barretzoph
10+
/research/autoencoders/ @snurkabill
11+
/research/brain_coder/ @danabo
12+
/research/cognitive_mapping_and_planning/ @s-gupta
13+
/research/compression/ @nmjohn
14+
/research/deep_contextual_bandits/ @rikel
15+
/research/deeplab/ @aquariusjay @yknzhu @gpapan
16+
/research/delf/ @andrefaraujo
17+
/research/differential_privacy/ @ilyamironov @ananthr
18+
/research/domain_adaptation/ @bousmalis @dmrd
19+
/research/efficient-hrl/ @ofirnachum
20+
/research/gan/ @joel-shor
21+
/research/global_objectives/ @mackeya-google
22+
/research/im2txt/ @cshallue
23+
/research/inception/ @shlens @vincentvanhoucke
24+
/research/keypointnet/ @mnorouzi
25+
/research/learned_optimizer/ @olganw @nirum
26+
/research/learning_to_remember_rare_events/ @lukaszkaiser @ofirnachum
27+
/research/learning_unsupervised_learning/ @lukemetz @nirum
28+
/research/lexnet_nc/ @vered1986 @waterson
29+
/research/lfads/ @jazcollins @susillo
30+
/research/lm_1b/ @oriolvinyals @panyx0718
31+
/research/lm_commonsense/ @thtrieu
32+
/research/marco/ @vincentvanhoucke
33+
/research/maskgan/ @a-dai
34+
/research/morph_net/ @gariel-google
35+
/research/namignizer/ @knathanieltucker
36+
/research/neural_gpu/ @lukaszkaiser
37+
/research/neural_programmer/ @arvind2505
38+
/research/next_frame_prediction/ @panyx0718
39+
/research/object_detection/ @jch1 @tombstone @derekjchow @jesu9 @dreamdragon @pkulzc
40+
/research/pcl_rl/ @ofirnachum
41+
/research/ptn/ @xcyan @arkanath @hellojas @honglaklee
42+
/research/real_nvp/ @laurent-dinh
43+
/research/rebar/ @gjtucker
44+
/research/resnet/ @panyx0718
45+
/research/seq2species/ @apbusia @depristo
46+
/research/skip_thoughts/ @cshallue
47+
/research/slim/ @sguada @nathansilberman
48+
/research/steve/ @buckman-google
49+
/research/street/ @theraysmith
50+
/research/swivel/ @waterson
51+
/research/syntaxnet/ @calberti @andorardo @bogatyy @markomernick
52+
/research/tcn/ @coreylynch @sermanet
53+
/research/tensorrt/ @karmel
54+
/research/textsum/ @panyx0718 @peterjliu
55+
/research/transformer/ @daviddao
56+
/research/vid2depth/ @rezama
57+
/research/video_prediction/ @cbfinn
58+
/research/fivo/ @dieterichlawson
59+
/samples/ @MarkDaoust @lamberta
60+
/samples/languages/java/ @asimshankar
61+
/tutorials/embedding/ @zffchen78 @a-dai
62+
/tutorials/image/ @sherrym @shlens
63+
/tutorials/image/cifar10_estimator/ @tfboyd @protoget
64+
/tutorials/rnn/ @lukaszkaiser @ebrevdo

ISSUE_TEMPLATE.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ http://stackoverflow.com/questions/tagged/tensorflow
44

55
Also, please understand that many of the models included in this repository are experimental and research-style code. If you open a GitHub issue, here is our policy:
66

7-
1. It must be a bug or a feature request.
7+
1. It must be a bug, a feature request, or a significant problem with documentation (for small docs fixes please send a PR instead).
88
2. The form below must be filled out.
99

1010
**Here's why we have that policy**: TensorFlow developers respond to issues. We want to focus on work that benefits the whole community, e.g., fixing bugs and adding features. Support only helps individuals. GitHub also notifies thousands of people when issues are filed. We want them to see you communicating an interesting problem, rather than being redirected to Stack Overflow.

README.md

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,19 @@
11
# TensorFlow Models
22

3-
This repository contains a number of different models implemented in [TensorFlow](https://tensorflow.org):
3+
This repository contains a number of different models implemented in [TensorFlow](https://www.tensorflow.org):
44

55
The [official models](official) are a collection of example models that use TensorFlow's high-level APIs. They are intended to be well-maintained, tested, and kept up to date with the latest stable TensorFlow API. They should also be reasonably optimized for fast performance while still being easy to read. We especially recommend newer TensorFlow users to start here.
66

7-
The [research models](research) are a large collection of models implemented in TensorFlow by researchers.
7+
The [research models](https://github.com/tensorflow/models/tree/master/research) are a large collection of models implemented in TensorFlow by researchers. They are not officially supported or available in release branches; it is up to the individual researchers to maintain the models and/or provide support on issues and pull requests.
88

9-
The [tutorial models](tutorials) are models described in the [TensorFlow tutorials](https://www.tensorflow.org/tutorials/).
9+
The [samples folder](samples) contains code snippets and smaller models that demonstrate features of TensorFlow, including code presented in various blog posts.
10+
11+
The [tutorials folder](tutorials) is a collection of models described in the [TensorFlow tutorials](https://www.tensorflow.org/tutorials/).
12+
13+
## Contribution guidelines
14+
15+
If you want to contribute to models, be sure to review the [contribution guidelines](CONTRIBUTING.md).
16+
17+
## License
18+
19+
[Apache License 2.0](LICENSE)

official/Dockerfile.cpu

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
# Docker image for running examples in Tensorflow models.
2+
# base_image depends on whether we are running on GPUs or non-GPUs
3+
FROM ubuntu:latest
4+
5+
RUN apt-get update && apt-get install -y --no-install-recommends \
6+
ca-certificates \
7+
build-essential \
8+
git \
9+
python \
10+
python-pip \
11+
python-setuptools
12+
13+
RUN pip install tf-nightly
14+
15+
# Checkout tensorflow/models at HEAD
16+
RUN git clone https://github.com/tensorflow/models.git /tensorflow_models
17+

official/Dockerfile.gpu

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
# Docker image for running examples in Tensorflow models.
2+
# base_image depends on whether we are running on GPUs or non-GPUs
3+
FROM nvidia/cuda:9.0-cudnn7-runtime-ubuntu16.04
4+
5+
RUN apt-get update && apt-get install -y --no-install-recommends \
6+
ca-certificates \
7+
build-essential \
8+
git \
9+
python \
10+
python-pip \
11+
python-setuptools
12+
13+
RUN pip install tf-nightly-gpu
14+
15+
# Checkout tensorflow/models at HEAD
16+
RUN git clone https://github.com/tensorflow/models.git /tensorflow_models
17+
18+

official/README.md

Lines changed: 57 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,66 @@
11
# TensorFlow Official Models
22

3-
The TensorFlow official models are a collection of example models that use TensorFlow's high-level APIs. They are intended to be well-maintained, tested, and kept up to date with the latest stable TensorFlow API. They should also be reasonably optimized for fast performance while still being easy to read.
3+
The TensorFlow official models are a collection of example models that use TensorFlow's high-level APIs. They are intended to be well-maintained, tested, and kept up to date with the latest TensorFlow API. They should also be reasonably optimized for fast performance while still being easy to read.
44

5-
Below is the list of models contained in the garden:
5+
These models are used as end-to-end tests, ensuring that the models run with the same speed and performance with each new TensorFlow build.
66

7-
[mnist](mnist): A basic model to classify digits from the MNIST dataset.
7+
## Tensorflow releases
8+
The master branch of the models are **in development**, and they target the [nightly binaries](https://github.com/tensorflow/tensorflow#installation) built from the [master branch of TensorFlow](https://github.com/tensorflow/tensorflow/tree/master). We aim to keep them backwards compatible with the latest release when possible (currently TensorFlow 1.5), but we cannot always guarantee compatibility.
89

9-
[resnet](resnet): A deep residual network that can be used to classify both CIFAR-10 and ImageNet's dataset of 1000 classes.
10+
**Stable versions** of the official models targeting releases of TensorFlow are available as tagged branches or [downloadable releases](https://github.com/tensorflow/models/releases). Model repository version numbers match the target TensorFlow release, such that [branch r1.4.0](https://github.com/tensorflow/models/tree/r1.4.0) and [release v1.4.0](https://github.com/tensorflow/models/releases/tag/v1.4.0) are compatible with [TensorFlow v1.4.0](https://github.com/tensorflow/tensorflow/releases/tag/v1.4.0).
1011

11-
[wide_deep](wide_deep): A model that combines a wide model and deep network to classify census income data.
12+
If you are on a version of TensorFlow earlier than 1.4, please [update your installation](https://www.tensorflow.org/install/).
1213

13-
More models to come!
14+
## Requirements
15+
Please follow the below steps before running models in this repo:
16+
17+
1. Add the top-level ***/models*** folder to the Python path with the command:
18+
```
19+
export PYTHONPATH="$PYTHONPATH:/path/to/models"
20+
```
21+
2. Install dependencies:
22+
```
23+
pip3 install --user -r official/requirements.txt
24+
```
25+
or
26+
```
27+
pip install --user -r official/requirements.txt
28+
```
29+
30+
31+
To make Official Models easier to use, we are planning to create a pip installable Official Models package. This is being tracked in [#917](https://github.com/tensorflow/models/issues/917).
32+
33+
34+
## Available models
35+
36+
**NOTE:** Please make sure to follow the steps in the [Requirements](#requirements) section.
37+
38+
* [boosted_trees](boosted_trees): A Gradient Boosted Trees model to classify higgs boson process from HIGGS Data Set.
39+
* [mnist](mnist): A basic model to classify digits from the MNIST dataset.
40+
* [resnet](resnet): A deep residual network that can be used to classify both CIFAR-10 and ImageNet's dataset of 1000 classes.
41+
* [transformer](transformer): A transformer model to translate the WMT English to German dataset.
42+
* [wide_deep](wide_deep): A model that combines a wide model and deep network to classify census income data.
43+
* More models to come!
1444

1545
If you would like to make any fixes or improvements to the models, please [submit a pull request](https://github.com/tensorflow/models/compare).
46+
47+
## New Models
48+
49+
The team is actively working to add new models to the repository. Every model should follow the following guidelines, to uphold the
50+
our objectives of readable, usable, and maintainable code.
51+
52+
**General guidelines**
53+
* Code should be well documented and tested.
54+
* Runnable from a blank environment with relative ease.
55+
* Trainable on: single GPU/CPU (baseline), multiple GPUs, TPU
56+
* Compatible with Python 2 and 3 (using [six](https://pythonhosted.org/six/) when necessary)
57+
* Conform to [Google Python Style Guide](https://github.com/google/styleguide/blob/gh-pages/pyguide.md)
58+
59+
**Implementation guidelines**
60+
61+
These guidelines exist so the model implementations are consistent for better readability and maintainability.
62+
63+
* Use [common utility functions](utils)
64+
* Export SavedModel at the end of training.
65+
* Consistent flags and flag-parsing library ([read more here](utils/flags/guidelines.md))
66+
* Produce benchmarks and logs ([read more here](utils/logs/guidelines.md))

official/__init__.py

Whitespace-only changes.

official/benchmark/__init__.py

Whitespace-only changes.

0 commit comments

Comments
 (0)