This repository provides a curated list of the GitHub repositories with machine learning models and implementations powered by TensorFlow 2.
Note: 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 want to contribute, please review the contribution guidelines.