Beckschen

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

sentence-transformers

Multilingual Sentence & Image Embeddings with BERT

taming-transformers

Taming Transformers for High-Resolution Image Synthesis

DALLE-pytorch

Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch

deit

Official DeiT repository

mmpretrain

OpenMMLab Pre-training Toolbox and Benchmark

medical-imaging-datasets

A list of Medical imaging datasets.

ganspace

Discovering Interpretable GAN Controls [NeurIPS 2020]

big_transfer

Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.

pytorch-seq2seq

An open source framework for seq2seq models in PyTorch.

interfacegan

[CVPR 2020] Interpreting the Latent Space of GANs for Semantic Face Editing

SETR

[CVPR 2021] Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers

sefa

[CVPR 2021] Closed-Form Factorization of Latent Semantics in GANs

centermask2

[CVPR 2020] CenterMask : Real-time Anchor-Free Instance Segmentation

point-transformer-pytorch

Implementation of the Point Transformer layer, in Pytorch

closed-form-matting

Python implementation of A. Levin D. Lischinski and Y. Weiss. A Closed Form Solution to Natural Image Matting. IEEE Conf. on Computer Vision and Pattern Recognition (CVPR), June 2006, New York

STAR

[ECCV 2020] Code for "Spatio-Temporal Graph Transformer Networks for Pedestrian Trajectory Prediction"

fairseq-image-captioning

Transformer-based image captioning extension for pytorch/fairseq

GAN-MRI

Code repository for Frontiers article 'Generative Adversarial Networks for Image-to-Image Translation on Multi-Contrast MR Images - A Comparison of CycleGAN and UNIT'

awesome-generative-modeling

Bolei's archive on generative modeling

GANLocalEditing

triq

TRIQ implementation

self-adj-dice

Implementation of Self-adjusting Dice Loss from "Dice Loss for Data-imbalanced NLP Tasks" paper

Unpaired_MR_to_CT_Image_Synthesis

Whole-body MR to CT image mapping

Medical-Image-Synthesis

It describes MRI to CT conversion, 3T to 7T conversion using Generative Adversarial Network (GAN).

awesome-instance-segmentation

High-res-disentanglement-datasets

Datasets for new state-of-the-art challenge in disentanglement learning

ImageTransformer

Image Captioning through Image Transformer

ARL

The pytorch implement for Attention Residual Learning for Skin Lesion Classification

ISIC

Two stage Framework for Skin Lesion Segmentation and Classification-- ISIC workshop