Model Zoo

This repository contains implementations of various deep learning research papers.

The models are broadly categorised into the folders Generative (e.g. various generative models), NLP (e.g. various recurrent neural networks (RNNs) and natural language processing (NLP) models), Classification (e.g. various CNN models to classify images), Object Detection, Multimodal , Super resolution , 3D Computer Vision, Semantic Segmentation. See the READMEs of respective models for more information.

Model Listing

• Generative models
  • AAE PyTorch
  • ACGAN PyTorch
  • CycleGAN PyTorch
  • DCGAN TensorFlow PyTorch
  • DiscoGAN PyTorch
  • InfoGAN TensorFlow PyTorch
  • MoCoGAN PyTorch
  • StarGAN PyTorch
  • VAEGAN PyTorch
  • VanillaGAN TensorFlow PyTorch
  • WGAN TensorFlow
  • cGAN TensorFlow PyTorch
  • SS-GAN TensorFlow
• NLP
  • BERT TensorFlow PyTorch
  • BI-LSTM_CRF TensorFlow
  • GloVe TensorFlow PyTorch
  • Transformer TensorFlow
  • Word2Vec PyTorch
• classification
  • ResNet PyTorch
  • ResNet56 Tensorflow
  • T3D ConvNets TensorFlow
  • Inception_V1 PyTorch Tensorflow
  • Inception_V3 PyTorch
  • Xception Tensorflow
  • MLP_Mixer PyTorch Tensorflow
  • MobileNet_V1 Tensorflow
  • I3D Tensorflow
  • RepVGG Tensorflow PyTorch
• Multimodal models
  • StackGAN PyTorch
  • VQA TensorFlow
  • WaveGAN TensorFlow
  • ImageCaption TensorFlow PyTorch
  • Show and Tell PyTorch Tensorflow
  • StackGAN_V1 Tensorflow
  • StackGAN_V2 PyTorch
• Semantic Segmentation
  • UNet Pytorch
  • UNet++ Pytorch
• Object detection
  • YOLOv2 TensorFlow
  • YOLOv3 TensorFlow
  • YOLOv1 Tensorflow
  • Faster_RCNN Tensorflow
  • SSD Tensorflow
• Super resolution
  • ContextEncoder TensorFlow PyTorch
  • SRGAN TensorFlow PyTorch
  • SRCNN TensorFlow
  • VDSR PyTorch
  • Perceptual Loss for real-time Style Transfer and Super Resolution PyTorch
• 3D Computer Vision
  • 3DGAN PyTorch