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 | Paper | Features | Maintainer |
|---|---|---|---|
| DenseNet 169 | Densely Connected Convolutional Networks | • FP32 Inference | Intel |
| Inception V3 | Rethinking the Inception Architecture for Computer Vision |
• Int8 Inference • FP32 Inference |
Intel |
| Inception V4 | Inception-v4, Inception-ResNet and the Impact of Residual Connections on Learning |
• Int8 Inference • FP32 Inference |
Intel |
| MobileNet V1 | MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications |
• Int8 Inference • FP32 Inference |
Intel |
| ResNet 101 | Deep Residual Learning for Image Recognition | • Int8 Inference • FP32 Inference |
Intel |
| ResNet 50 | Deep Residual Learning for Image Recognition | • Int8 Inference • FP32 Inference |
Intel |
| ResNet 50v1.5 | Deep Residual Learning for Image Recognition | • Int8 Inference • FP32 Inference • FP32 Training |
Intel |
| Model | 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.