This repository provides a curated list of the GitHub repositories with machine learning models and implementations powered by TensorFlow 2.
Disclosure: Contributing companies or individuals are responsible for maintaining their repositories.
| Model | Reference (Paper) | Features | Maintainer |
|---|---|---|---|
| Mask R-CNN | Mask R-CNN | • Automatic Mixed Precision • Multi-GPU training support with Horovod • TensorRT |
NVIDIA |
| U-Net Medical Image Segmentation | U-Net: Convolutional Networks for Biomedical Image Segmentation | • Automatic Mixed Precision • Multi-GPU training support with Horovod • TensorRT |
NVIDIA |
If you have a suggestion for the community model repository, please submit a pull request.