ExplainingAI's repositories
StableDiffusion-PyTorch
This repo implements a Stable Diffusion model in PyTorch with all the essential components.
DDPM-Pytorch
This repo implements Denoising Diffusion Probabilistic Models (DDPM) in Pytorch
VQVAE-Pytorch
This repo implements VQVAE on mnist and as well as colored version of mnist images. It also implements simple LSTM for generating sample numbers using the encoder outputs of trained VQVAE
DiT-PyTorch
This repo implements Diffusion Transformers(DiT) in PyTorch and provides training and inference code on CelebHQ dataset
FasterRCNN-PyTorch
This repo implements simple Faster RCNN model in PyTorch with all the essential components.
VAE-Pytorch
This repository implements a simpleVAE for training on CPU on the MNIST dataset and provides ability to visualize the latent space, entire manifold as well as visualize how numbers interpolate between each other.
ControlNet-PyTorch
This repo implements ControlNet with DDPM and Latent Diffusion Model in PyTorch with canny edges as conditional control for Mnist and CelebHQ
VIT-Pytorch
This repo implements and trains Vision Transformer (VIT) on a synthetically generated dataset which has colored mnist images on texture backgrounds
SSD-PyTorch
This repo implements SSD in PyTorch and provides training and inference code on VOC dataset
VideoGeneration-PyTorch
This repo implements Video generation model using Latent Diffusion Transformers(Latte) in PyTorch and provides training and inference code on Moving mnist dataset and UCF101 dataset
Dalle-Pytorch
This repo implements and trains DallE-1 on a synthetically generated dataset which has colored mnist images on texture/solid background and auto generated captions.
Yolov1-PyTorch
This repo implements Yolov1 in PyTorch and provides training and inference code on VOC dataset
DETR-PyTorch
This repo implements DETR in PyTorch and provides training and inference code on VOC dataset
Normalizing-Flow
This repo implements normalizing flow model with a simple realnvp like model and provides training and sampling code on mnist dataset
GANs-Pytorch
This repo implements a simple GAN with fc layers and trains it on MNIST
DCGAN-Pytorch
This repo implements DCGAN model and trains it on MNIST and Celeb Faces dataset
DetectionUtils
This repository provides implementation of utilities for Object Detection