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
- pytorch
- resnet
- pretrained-models
- pretrained-weights
Matlab/Octave toolbox for deep learning. Includes Deep Belief Nets, Stacked Autoencoders, Convolutional Neural Nets, Convolutional Autoencoders and vanilla Neural Nets. Each method has examples to get you started.
[CVPR2020] Adversarial Latent Autoencoders
This repository includes the official project of TransUNet, presented in our paper: TransUNet: Transformers Make Strong Encoders for Medical Image Segmentation.
Unofficial PyTorch implementation of Masked Autoencoders Are Scalable Vision Learners
A library of sklearn compatible categorical variable encoders
Official PyTorch Implementation of "Diffusion Transformers with Representation Autoencoders"
[CVPR'24 Oral] Official repository of Point Transformer V3 (PTv3). Also a pip-installable PyTorch + Triton package with pretrained Sonata, Concerto, and Utonia encoders.
[NeurIPS 2022 Spotlight] VideoMAE: Masked Autoencoders are Data-Efficient Learners for Self-Supervised Video Pre-Training
- self-supervised-learning
- action-recognition
- video-understanding
- masked-autoencoder
Implementation of Graph Auto-Encoders in TensorFlow
Code for 'LLM2Vec: Large Language Models Are Secretly Powerful Text Encoders'
TorchXRayVision: A library of chest X-ray datasets and models. Classifiers, segmentation, and autoencoders.
Deep and Machine Learning for Microscopy
Official implementation of JAMIE, Joint variational Autoencoders for Multimodal Imputation and Embedding
- autoencoder
- imputation
- integration
- multimodal