shakediel

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

vit-pytorch

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

Ultra-Light-Fast-Generic-Face-Detector-1MB

💎1MB lightweight face detection model (1MB轻量级人脸检测模型)

optuna

A hyperparameter optimization framework

rlpyt

Reinforcement Learning in PyTorch

pytorch-for-numpy-users

PyTorch for Numpy users. https://pytorch-for-numpy-users.wkentaro.com

DeepSpeed

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

fiftyone

The open-source tool for building high-quality datasets and computer vision models

dtan

Official PyTorch implementation for our NeurIPS 2019 paper, Diffeomorphic Temporal Alignment Nets. TensorFlow\Keras version is available at tf_legacy branch.

deep-person-reid

Torchreid: Deep learning person re-identification in PyTorch.

tide

A General Toolbox for Identifying Object Detection Errors

fast-reid

SOTA Re-identification Methods and Toolbox

reid-strong-baseline

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

faiss

A library for efficient similarity search and clustering of dense vectors.

imgaug

Image augmentation for machine learning experiments.

pixel2style2pixel

Official Implementation for "Encoding in Style: a StyleGAN Encoder for Image-to-Image Translation"

SAM

Official Implementation for "Only a Matter of Style: Age Transformation Using a Style-Based Regression Model" (SIGGRAPH 2021) https://arxiv.org/abs/2102.02754

netron

Visualizer for neural network, deep learning and machine learning models

PlotNeuralNet

Latex code for making neural networks diagrams

vision_transformer

pixel2style2pixel

Official Implementation for "Encoding in Style: a StyleGAN Encoder for Image-to-Image Translation" (CVPR 2021) presenting the pixel2style2pixel (pSp) framework

pretrained-models.pytorch

Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.

react-vis-timeline

React component for the vis-timeline module

pix2vertex.pytorch

An official pyTorch port of the pix2vertex paper from ICCV2017

SimCLR

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations

Yet-Another-EfficientDet-Pytorch

The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.

json-config-expander

Expand multi optional configuration to multiple configurations.

awesome-community-detection

A curated list of community detection research papers with implementations.

react-image-gallery

React carousel image gallery component with thumbnail support 🖼

react-awesome-slider

React content transition slider. Awesome Slider is a 60fps, light weight, performant component that renders an animated set of production ready UI general purpose sliders with fullpage transition support for NextJS and GatsbyJS. 🖥️ 📱