rwightman

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)

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

genalog

Genalog is an open source, cross-platform python package allowing generation of synthetic document images with custom degradations and text alignment capabilities.

timm

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

imagenet-12k

ImageNet-12k subset of ImageNet-21k (fall11)

img2dataset

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

open_clip

An open source implementation of CLIP.

obstacle-tower-pytorch-rainbow

PyTorch & Rainbow for Obstacle Tower Challenge

webdataset

A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.

automl

Google Brain AutoML

huggingface_hub

All the open source things related to the Hugging Face Hub.

mmdetection

Open MMLab Detection Toolbox and Benchmark

pytorch

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

optax

Optax is a gradient processing and optimization library for JAX.

androids-dream

deep-neuroevolution

Deep Neuroevolution

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

poet

Paired Open-Ended Trailblazer (POET) and Enhanced POET

api-inference-community

CLIP_benchmark

CLIP-like model evaluation

phenaki

A phenaki reproduction using pytorch.