hyh12

Co-Mixup

Official PyTorch implementation of "Co-Mixup: Saliency Guided Joint Mixup with Supermodular Diversity" (ICLR'21 Oral)

classifier-balancing

This repository contains code for the paper "Decoupling Representation and Classifier for Long-Tailed Recognition", published at ICLR 2020

fpn.pytorch

Pytorch implementation of Feature Pyramid Network (FPN) for Object Detection

PixPro

Propagate Yourself: Exploring Pixel-Level Consistency for Unsupervised Visual Representation Learning, CVPR 2021

pytorch-grad-cam

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.

ShapeTextureDebiasedTraining

Code and models for the paper Shape-Texture Debiased Neural Network Training (ICLR 2021)

einops

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

vit-pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

trgan

Unsupervised Image Transformation Learning via Generative Adversarial Networks

xiaobai_fabu

小白加速全球唯一发布地址

torchdiffeq

Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.

ffjord

code for "FFJORD: Free-form Continuous Dynamics for Scalable Reversible Generative Models".

awesome-implicit-neural-models

mixup-cifar10

mixup: Beyond Empirical Risk Minimization

AI-Expert-Roadmap

Roadmap to becoming an Artificial Intelligence Expert in 2022

DALI_pytorch_demo

Example code showing how to use Nvidia DALI in pytorch, with fallback to torchvision. Contains a few differences to the official Nvidia example, namely a completely CPU pipeline & improved memory usage

DataLoaders_DALI

PyTorch DataLoaders implemented with DALI for accelerating image preprocessing

learning-invariances

Codebase for Learning Invariances in Neural Networks

Self-Attention-GAN

Pytorch implementation of Self-Attention Generative Adversarial Networks (SAGAN)

lambda-networks

Implementation of LambdaNetworks, a new approach to image recognition that reaches SOTA with less compute

lightning-bolts

Toolbox of models, callbacks, and datasets for AI/ML researchers.

reid-strong-baseline

Bag of Tricks and A Strong Baseline for Deep Person Re-identification

task_adaptation

DALI

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

TwoStageVAE

swav

PyTorch implementation of SwAV https//arxiv.org/abs/2006.09882

moco

PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722

PyTorch-VAE

A Collection of Variational Autoencoders (VAE) in PyTorch.

pytorch-lightning

Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.

amr

Official adversarial mixup resynthesis repository