ShivamShrirao

diffusers

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch

Dreambooth-Stable-Diffusion

Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) with Stable Diffusion (tweaks focused on training faces)

facegan_pytorch

Generate 256x256, 512x512 resolution images with simple Convolutional GAN by adding Gaussian noise to discriminator layers.

3D_res_attn_Unet

A custom 3D Unet style network with Residual Attention blocks and self attention for semantic segmentation. (in development)

shivamshrirao.github.io

VITON-HD

Official PyTorch implementation of "VITON-HD: High-Resolution Virtual Try-On via Misalignment-Aware Normalization" (CVPR 2021)

clip_classifier

pix2pixHD

Synthesizing and manipulating 2048x1024 images with conditional GANs

semantic-segmentation

SOTA Semantic Segmentation Models in PyTorch

animated_splash_screen_kde

CIHP_PGN

Code repository for Part Grouping Network, ECCV 2018

contrastive-unpaired-translation

Contrastive unpaired image-to-image translation, faster and lighter training than cyclegan (ECCV 2020, in PyTorch)

GANs_TF_2.0

DCGAN implementation in Tensorflow 2.

Image-Captioning-Transformers

instant-ngp

MIRNet

[ECCV 2020] Learning Enriched Features for Real Image Restoration and Enhancement. SOTA results for image denoising, super-resolution, and image enhancement.

nerf-pytorch

OneFormer

[CVPR 2023] OneFormer: One Transformer to Rule Universal Image Segmentation

py-thin-plate-spline

Code for computing interpolating / approximating thin plate splines.

pypotrace

Python bindings for potrace library, with to_xml (svg) converter by gungoren.

regnets-pytorch

stats-basic-notes

stylegan2-ada-pytorch

StyleGAN2-ADA - Official PyTorch implementation

TensorRT

tf2-to-torch-u2net

transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

U-2-Net

The code for our newly accepted paper in Pattern Recognition 2020: "U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection."

u2net_mobile

Vision-Transformer

xformers-wheels