Folders and files

Name		Name	Last commit message	Last commit date
Latest commit History 9,668 Commits
.github		.github
community		community
docs		docs
official		official
orbit		orbit
research		research
tensorflow_models		tensorflow_models
.gitignore		.gitignore
AUTHORS		AUTHORS
CODEOWNERS		CODEOWNERS
CODE_OF_CONDUCT.md		CODE_OF_CONDUCT.md
CONTRIBUTING.md		CONTRIBUTING.md
ISSUES.md		ISSUES.md
LICENSE		LICENSE
README.md		README.md
SECURITY.md		SECURITY.md

Repository files navigation

Welcome to the Model Garden for TensorFlow

The TensorFlow Model Garden is a repository with a number of differentimplementations of state-of-the-art (SOTA) models and modeling solutions forTensorFlow users. We aim to demonstrate the best practices for modeling so thatTensorFlow users can take full advantage of TensorFlow for their research andproduct development.

To improve the transparency and reproducibility of our models, training logs onTensorBoard.dev are also provided for models to theextent possible though not all models are suitable.

Directory	Description
official	• A collection of example implementations for SOTA models using the latest TensorFlow 2's high-level APIs • Officially maintained, supported, and kept up to date with the latest TensorFlow 2 APIs by TensorFlow • Reasonably optimized for fast performance while still being easy to read For more details on the capabilities, check the guide on theModel-garden
research	• A collection of research model implementations in TensorFlow 1 or 2 by researchers • Maintained and supported by researchers
community	• A curated list of the GitHub repositories with machine learning models and implementations powered by TensorFlow 2
orbit	• A flexible and lightweight library that users can easily use or fork when writing customized training loop code in TensorFlow 2.x. It seamlessly integrates with`tf.distribute` and supports running on different device types (CPU, GPU, and TPU).

Installation

To install the current release of tensorflow-models, please follow any one of the methods described below.

Method 1: Install the TensorFlow Model Garden pip package

tf-models-official is the stable Model Garden package. Please check out thereleases to see what are available modules.

pip3 will install all models and dependencies automatically.

pip3 install tf-models-official

Please check out our examples:

basic library import
nlp model buildingto learn how to use a PIP package.

Note thattf-models-official may not include the latest changes in the master branch of thisgithub repo. To include latest changes, you may installtf-models-nightly,which is the nightly Model Garden package created daily automatically.

pip3 install tf-models-nightly

Method 2: Clone the source

Clone the GitHub repository:

git clone https://github.com/tensorflow/models.git

Add the top-level/models folder to the Python path.

export PYTHONPATH=$PYTHONPATH:/path/to/models

If you are using in a Windows environment, you may need to use the following command with PowerShell:

$env:PYTHONPATH +=":\path\to\models"

If you are using a Colab notebook, please set the Python path with os.environ.

importosos.environ['PYTHONPATH']+=":/path/to/models"

Install other dependencies

pip3 install --user -r models/official/requirements.txt

Finally, if you are using nlp packages, please also installtensorflow-text-nightly:

pip3 install tensorflow-text-nightly

Announcements

Please checkthis page for recent announcements.

Contributions

If you want to contribute, please review thecontribution guidelines.

License

Apache License 2.0

Citing TensorFlow Model Garden

If you use TensorFlow Model Garden in your research, please cite this repository.

@misc{tensorflowmodelgarden2020,  author = {Hongkun Yu and Chen Chen and Xianzhi Du and Yeqing Li and Abdullah Rashwan and Le Hou and Pengchong Jin and Fan Yang            and Frederick Liu and Jaeyoun Kim and Jing Li},  title = {{TensorFlow Model Garden}},  howpublished = {\url{https://github.com/tensorflow/models}},  year = {2020}}