lithostark

GNN-Resource-Management

ACE

Code for our ICML '19 paper: Neural Network Attributions: A Causal Perspective.

Variational-Capsule-Routing

Official Pytorch code for (AAAI 2020) paper "Capsule Routing via Variational Bayes", https://arxiv.org/pdf/1905.11455.pdf

pytorch-mobilenet-v3

MobileNetV3 in pytorch and ImageNet pretrained models

NBD_KerUnc

Project page of the paper 'Deep Learning for Handling Kernel/model Uncertainty in Image Deconvolution' (CVPR 2020)

flatnet

This is the official pytorch code repo for the ICCV 2019 paper Towards Photorealistic Reconstruction of Highly Multiplexed Lensless Images and the TPAMI 2020 paper FlatNet: Towards Photorealistic Scene Reconstruction from Lensless Measurements

pytorch-CycleGAN-and-pix2pix

Image-to-Image Translation in PyTorch

set_transformer

Pytorch implementation of set transformer

PyTorch-GAN

PyTorch implementations of Generative Adversarial Networks.

DGG

Linxiao Yang, Ngai-Man Cheung, Jiaying Li, and Jun Fang, "Deep Clustering by Gaussian Mixture Variational Autoencoders with Graph Embedding", In ICCV 2019.

mine-pytorch

Mutual Information Neural Estimation in Pytorch

torch-scae

PyTorch implementation of Stacked Capsule Auto-Encoders

SimCLR

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations by T. Chen et al.

Transformation-CR

Transformation Consistency Regularization- A Semi-Supervised Paradigm for Image-to-Image Translation

pytorch_image_classification

PyTorch implementation of image classification models for CIFAR-10/CIFAR-100/MNIST/FashionMNIST/Kuzushiji-MNIST/ImageNet

pytorch-playground

Base pretrained models and datasets in pytorch (MNIST, SVHN, CIFAR10, CIFAR100, STL10, AlexNet, VGG16, VGG19, ResNet, Inception, SqueezeNet)

fast-autoaugment

Fast AutoAugment, similar to official release but easier to run

Wide-resnet-fashionMnist

wide resnet for fashion mnist

densenet-svhn-classification-pytorch

fast-autoaugment

Official Implementation of 'Fast AutoAugment' in PyTorch.

wide-residual-networks

3.8% and 18.3% on CIFAR-10 and CIFAR-100

DIVA

Implementation of 'DIVA: Domain Invariant Variational Autoencoders'

SupContrast

PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)

DomainBed

DomainBed is a suite to test domain generalization algorithms

NetVLAD-pytorch

PyTorch implementation of NetVLAD & Online Hardest Triplet Loss.

computer_vision_experiments_loss_functions_tests

Comparison between Contrastive, Triplet and Yukawa loss functions

wide-resnet.pytorch

Best CIFAR-10, CIFAR-100 results with wide-residual networks using PyTorch

Fisher-Triplet-Contrastive-Loss

The code for Fisher Discriminant Triplet (FDT) and Fisher Discriminant Contrastive (FDC) loss functions

Xlearn

Transfer Learning Library

dg_mmld

Pytorch Implementation of Domain Generalization Using a Mixture of Multiple Latent Domains