Wayne2Wang

MinkowskiEngine

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors

convolution-visualizer

Convolution visualizations

FastFCN

FastFCN: Rethinking Dilated Convolution in the Backbone for Semantic Segmentation.

regnet

Pytorch implementation of network design paradigm described in the paper "Designing Network Design Spaces"

pycls

Codebase for Image Classification Research, written in PyTorch.

nerf

Code release for NeRF (Neural Radiance Fields)

fast-neural-style

Feedforward style transfer

numpy

The fundamental package for scientific computing with Python.

the-gan-zoo

A list of all named GANs!

pytorch3d

PyTorch3D is FAIR's library of reusable components for deep learning with 3D data

webdataset

A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.

pytorch-fid

Compute FID scores with PyTorch.

stylegan

StyleGAN - Official TensorFlow Implementation

sngan_projection

GANs with spectral normalization and projection discriminator

simple-pytorch-3dgan

A simple and unofficial 3D-GAN implementation using PyTorch [NeurIPS 2016]

3DGAN-Pytorch

3DGAN-Pytorch

ganhacks

starter from "How to Train a GAN?" at NIPS2016

Pix2Pix-for-Semantic-Segmentation-of-Satellite-Images

The presented experiment aims at using Pix2Pix network to segment the building footprint from Satellite Images.

3D-GAN-pytorch

Responsible implementation of 3D-GAN NIPS 2016 paper:Learning a Probabilistic Latent Space of ObjectShapes via 3D Generative-Adversarial Modeling,that can be found https://papers.nips.cc/paper/6096-learning-a-probabilistic-latent-space-of-object-shapes-via-3d-generative-adversarial-modeling.pdf

torchio

Medical imaging toolkit for deep learning

tinyrenderer

A brief computer graphics / rendering course

cityscapesScripts

README and scripts for the Cityscapes Dataset

Pytorch-UNet

PyTorch implementation of the U-Net for image semantic segmentation with high quality images

PyTorch-GAN

PyTorch implementations of Generative Adversarial Networks.

3DStyleGAN

3D StyleGAN2 for Medical Images

nnUNet

torchdistill

A coding-free framework built on PyTorch for reproducible deep learning studies. 🏆25 knowledge distillation methods presented at CVPR, ICLR, ECCV, NeurIPS, ICCV, etc are implemented so far. 🎁 Trained models, training logs and configurations are available for ensuring the reproducibiliy and benchmark.

pytorch-image-models

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

ffcv

FFCV: Fast Forward Computer Vision (and other ML workloads!)