rwightman
Always learning, constantly curious. Building ML/AI systems, watching loss curves.
Vancouver, BC
Pinned Repositories
pytorch-image-models
PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNet-V3/V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more
efficientdet-pytorch
A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights
efficientnet-jax
EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
gen-efficientnet-pytorch
Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS
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
pytorch-nips2017-attack-example
A PyTorch baseline attack example for the NIPS 2017 adversarial competition
udacity-driving-reader
Quick docker based reader for udacity rosbag self-driving dataset. Dumps to png/jpg + csv or Tensorflow examples.
rwightman's Repositories
rwightman/gen-efficientnet-pytorch
Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS
rwightman/efficientdet-pytorch
A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights
rwightman/posenet-python
A Python port of Google TensorFlow.js PoseNet (Real-time Human Pose Estimation)
rwightman/posenet-pytorch
A PyTorch port of Google TensorFlow.js PoseNet (Real-time Human Pose Estimation)
rwightman/udacity-driving-reader
Quick docker based reader for udacity rosbag self-driving dataset. Dumps to png/jpg + csv or Tensorflow examples.
rwightman/efficientnet-jax
EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
rwightman/genalog
Genalog is an open source, cross-platform python package allowing generation of synthetic document images with custom degradations and text alignment capabilities.
rwightman/timm
PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more
rwightman/imagenet-12k
ImageNet-12k subset of ImageNet-21k (fall11)
rwightman/img2dataset
Easily turn large sets of image urls to an image dataset. Can download, resize and package 100M urls in 20h on one machine.
rwightman/open_clip
An open source implementation of CLIP.
rwightman/obstacle-tower-pytorch-rainbow
PyTorch & Rainbow for Obstacle Tower Challenge
rwightman/webdataset
A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.
rwightman/automl
Google Brain AutoML
rwightman/huggingface_hub
All the open source things related to the Hugging Face Hub.
rwightman/mmdetection
Open MMLab Detection Toolbox and Benchmark
rwightman/pytorch
rwightman/vision
Datasets, Transforms and Models specific to Computer Vision
rwightman/kinetics-download-prep
Download kinetics and prepare '.json' annotations for each subset
rwightman/maskrcnn-benchmark
Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.
rwightman/objax
rwightman/optax
Optax is a gradient processing and optimization library for JAX.
rwightman/androids-dream
rwightman/deep-neuroevolution
Deep Neuroevolution
rwightman/detectron2
Detectron2 is FAIR's next-generation platform for object detection and segmentation.
rwightman/graphcore-examples
Example code and applications for machine learning on Graphcore IPUs
rwightman/poet
Paired Open-Ended Trailblazer (POET) and Enhanced POET
rwightman/api-inference-community
rwightman/CLIP_benchmark
CLIP-like model evaluation
rwightman/phenaki
A phenaki reproduction using pytorch.