Ross Wightman's repositories
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)
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
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.
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.
huggingface_hub
All the open source things related to the Hugging Face Hub.
mmdetection
Open MMLab Detection Toolbox and Benchmark
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.
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
CLIP_benchmark
CLIP-like model evaluation