Below is a list of models that we would like to have in the model zoo in the near future. Please refer to the contribution guidelines to contribute a model.
| Model | Reference |
|---|---|
| Gender Detection | Age and Gender Classification using Convolutional Neural Networks |
| Single Shot Detector | SSD: Single Shot Multi Detector |
| Super Resolution | Image Super resolution using deep convolutional networks |
| Face Detection | A Convolutional Neural Network Cascade for Face Detection |
| Face Detection | ArcFace: Additive Angular Margin Loss for Deep Face Recognition |
| Semantic Segmentation | Fully Convolutional Networks for Semantic Segmentation |
| Object Detection & Segmentation | Faster-RCNN |
| Object Detection & Segmentation | Mask-RCNN |
| Object Detection & Segmentation | YOLO v2 / v3 |
| Model | Reference |
|---|---|
| Text to Image | Generative Adversarial Text to image Synthesis |
| Style Transfer | Unpaired Image to Image Translation using Cycle consistent Adversarial Network |
| Sound Generative models | WaveNet: A Generative Model for Raw Audio |
| Model | Reference |
|---|---|
| Speech Recognition | Speech recognition with deep recurrent neural networks |
| Text To Speech | Deep voice: Real time neural text to speech |
| Language Model | Deep Neural Network Language Models |
| Machine Translation | Neural Machine Translation by jointly learning to align and translate |
| Machine Translation | Google's Neural Machine Translation System: Bridging the Gap between Human and Machine Translation |
| Model | Reference |
|---|---|
| Visual Question Answering | VQA: Visual Question Answering |
| Visual Question Answering | Yin and Yang: Balancing and Answering Binary Visual Questions |
| Visual Question Answering | Making the V in VQA Matter: Elevating the Role of Image Understanding in Visual Question Answering |
| Visual Dialog | Visual Dialog |
| Model | Reference |
|---|---|
| Time Series Forecasting | Modeling Long- and Short-Term Temporal Patterns with Deep Neural Networks |
| Recommender systems | DropoutNet: Addressing Cold Start in Recommender Systems |
| Collaborative filtering | |
| Autoencoders |
Want other models in this Model Zoo? Go ahead and raise a GitHub issue outlining details and update the above list to call the community for contribution.