rwightman
Computer Vision @huggingface. Always learning, constantly curious. Building ML/AI systems, watching loss curves.
Repositories
Select a repository to view its commits, contributors, and more.efficientdet-pytorch
A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights
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
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-nips2017-attack-example
A PyTorch baseline attack example for the NIPS 2017 adversarial competition
tensorflow-litterbox
Tensorflow experimentation sandbox. VGG, ResNet, Inception V3, Inception V4, Inception-Resnet-V2 models.
pytorch-pommerman-rl
PyTorch RL for Pommerman
pytorch-planet-amazon
PyTorch models and training code for 'Planet: Understanding the Amazon from Space' Kaggle
imagenet-12k
ImageNet-12k subset of ImageNet-21k (fall11)