rwightman

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

gen-efficientnet-pytorch

Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS

efficientdet-pytorch

A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights

posenet-python

A Python port of Google TensorFlow.js PoseNet (Real-time Human Pose Estimation)

posenet-pytorch

A PyTorch port of Google TensorFlow.js PoseNet (Real-time Human Pose Estimation)

pytorch-dpn-pretrained

Dual Path Networks (DPN) supporting pretrained weights converted from original MXNet implementation

udacity-driving-reader

Quick docker based reader for udacity rosbag self-driving dataset. Dumps to png/jpg + csv or Tensorflow examples.

efficientnet-jax

EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax

pytorch-pretrained-gluonresnet

Well trained MXNet Gluon Model Zoo ResNet/ResNeXt/SE-ResNeXt ported to PyTorch

pytorch-commands

Some PyTorch code for the Kaggle Speech Recognition Challenge

pytorch-opensim-rl

PyTorch based Reinforcement Learning for OpenSim Prosthetics and Learning to Run environments

img2dataset

Easily turn large sets of image urls to an image dataset. Can download, resize and package 100M urls in 20h on one machine.

obstacle-tower-pytorch-rainbow

PyTorch & Rainbow for Obstacle Tower Challenge

open_clip

An open source implementation of CLIP.

automl

Google Brain AutoML

vision

Datasets, Transforms and Models specific to Computer Vision

kinetics-download-prep

Download kinetics and prepare '.json' annotations for each subset

maskrcnn-benchmark

Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.

objax

obstacle-tower-pytorch-a2c-ppo

PPO/A2C in PyTorch for the Obstacle Tower Challenge

optax

Optax is a gradient processing and optimization library for JAX.

pytorch

androids-dream

deep-neuroevolution

Deep Neuroevolution

DensePose

A real-time approach for mapping all human pixels of 2D RGB images to a 3D surface-based model of the body

detectron2

Detectron2 is FAIR's next-generation platform for object detection and segmentation.

graphcore-examples

Example code and applications for machine learning on Graphcore IPUs

mmdetection

Open MMLab Detection Toolbox and Benchmark

poet

Paired Open-Ended Trailblazer (POET) and Enhanced POET

pytorch-cdiscount