Welcome to the Model Garden for TensorFlow

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

To improve the transparency and reproducibility of our models, training logs on TensorBoard.dev are also provided for models to the extent 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 the Model-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 the releases 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 building to learn how to use a PIP package.

Note that tf-models-official may not include the latest changes in the master branch of this github repo. To include latest changes, you may install tf-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.

import os
os.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 install tensorflow-text-nightly:

pip3 install tensorflow-text-nightly

Announcements

Please check this page for recent announcements.

Contributions

If you want to contribute, please review the contribution 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, Chen Chen, Xianzhi Du, Yeqing Li, Abdullah Rashwan, Le Hou, Pengchong Jin, Fan Yang,
            Frederick Liu, Jaeyoun Kim, and Jing Li},
  title = {{TensorFlow Model Garden}},
  howpublished = {\url{https://github.com/tensorflow/models}},
  year = {2020}
}

Project Name	Stars	Repos Using This	Packages Using This	Most Recent Commit	Total Releases	Latest Release	Open Issues	License	Language
Models	75,899		26	2 days ago	23	May 20, 2022	1,205	other	Jupyter Notebook
Models and examples built with TensorFlow
Segmentation_models	4,095	12	12	7 months ago	8	January 10, 2020	237	mit	Python
Segmentation models with pretrained backbones. Keras and TensorFlow Keras.
Keras Unet Collection	428		3	4 months ago	32	January 10, 2022	28	mit	Python
The Tensorflow, Keras implementation of U-net, V-net, U-net++, UNET 3+, Attention U-net, R2U-net, ResUnet-a, U^2-Net, TransUNET, and Swin-UNET with optional ImageNet-trained backbones.
Tensorflow Serving Yolov3	358			2 years ago			28	mit	Python
本项目主要对原tensorflow-yolov3版本做了许多细节上的改进,增加了TensorFlow-Serving工程部署,训练了多个数据集，包括Visdrone2019, 安全帽等, 安全帽mAP在98%左右, 推理速度1080上608的尺寸大概25fps.
Tf.fashionai	239			5 years ago			3	apache-2.0	Python
Full pipeline for TianChi FashionAI clothes keypoints detection compitetion in TensorFlow
Tensorflow Advanced Segmentation Models	125			a month ago			8	other	Python
A Python Library for High-Level Semantic Segmentation Models based on TensorFlow and Keras with pretrained backbones.
X Detector	120			5 years ago			5	apache-2.0	Python
Light-Head RCNN and One Novel Object Detector
Ssd_efficientnet	57			4 years ago			11		Python
SSD using TensorFlow object detection API with EfficientNet backbone
Image Segmentation	35			4 years ago			2	mit	Python
Mask R-CNN, FPN, LinkNet, PSPNet and UNet with multiple backbone architectures support readily available
Smoke Detection Using Tensorflow 2.2	31			2 years ago			1	mit	Jupyter Notebook
An experimental repository to build ML models and perform efficient wildfire smoke detection.

Models