albertipot Goto Github PK

pycls

Codebase for Image Classification Research, written in PyTorch.

pygcn

Graph Convolutional Networks in PyTorch

pytorch-a2c-ppo-acktr-gail

PyTorch implementation of Advantage Actor Critic (A2C), Proximal Policy Optimization (PPO), Scalable trust-region method for deep reinforcement learning using Kronecker-factored approximation (ACKTR) and Generative Adversarial Imitation Learning (GAIL).

pytorch-fgvc-dataset

PyTorch custom dataset APIs -- CUB-200-2011, Stanford Dogs, Stanford Cars, FGVC Aircraft, NABirds, Tiny ImageNet, iNaturalist2017

pytorch-gan

PyTorch implementations of Generative Adversarial Networks.

pytorch-image-models

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more

pytorch-pcgrad

Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"

query2labels

Official implementation of paper "Query2Label: A Simple Transformer Way to Multi-Label Classification".

semantic-darts

semantic_nerf

The implementation of "In-Place Scene Labelling and Understanding with Implicit Scene Representation" [ICCV 2021].

sgnas

[CVPR 2021] 'Searching by Generating: Flexible and Efficient One-Shot NAS with Architecture Generator'

singlepathoneshot

slambook2

edition 2 of the slambook

slidr

Official PyTorch implementation of "Image-to-Lidar Self-Supervised Distillation for Autonomous Driving Data"

slimmable_networks

Slimmable Networks, AutoSlim, and Beyond, ICLR 2019, and ICCV 2019

smoke

SMOKE: Single-Stage Monocular 3D Object Detection via Keypoint Estimation

stable-diffusion

A latent text-to-image diffusion model

stanford-cars

Learning computer vision by striving to maximise accuracy on the Stanford Cars dataset

swin-transformer

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

swin-transformer-semantic-segmentation

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" on Semantic Segmentation.

t-nas

Source code of ICLR2020 paper 'Towards Fast Adaptation of Neural Architectures with Meta Learning'

tinygpt-v

TinyGPT-V: Efficient Multimodal Large Language Model via Small Backbones

tinyllama

The TinyLlama project is an open endeavor to pretrain a 1.1B Llama model on 3 trillion tokens.

ttt-lm-pytorch

Official PyTorch implementation of Learning to (Learn at Test Time): RNNs with Expressive Hidden States

unilm

Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities

unnas

Code for "Are labels necessary for neural architecture search"

vissl

VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.

vit-pytorch

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

vts-drloc

NeurIPS 2021, Official codes for "Efficient Training of Visual Transformers with Small Datasets".

yacs

YACS -- Yet Another Configuration System