ptrblck's starred 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
maskrcnn-benchmark
Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.
NYU-DLSP20
NYU Deep Learning Spring 2020
deep-learning-v2-pytorch
Projects and exercises for the latest Deep Learning ND program https://www.udacity.com/course/deep-learning-nanodegree--nd101
tensorwatch
Debugging, monitoring and visualization for Python Machine Learning and Data Science
loss-landscape
Code for visualizing the loss landscape of neural nets
pytorch-kaldi
pytorch-kaldi is a project for developing state-of-the-art DNN/RNN hybrid speech recognition systems. The DNN part is managed by pytorch, while feature extraction, label computation, and decoding are performed with the kaldi toolkit.
python-nubia
A command-line and interactive shell framework.
pytorch-ssd
MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in Pytorch 1.0 / Pytorch 0.4. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.
sklearn-porter
Transpile trained scikit-learn estimators to C, Java, JavaScript and others.
lightning-thunder
Make PyTorch models up to 40% faster! Thunder is a source to source compiler for PyTorch. It enables using different hardware executors at once; across one or thousands of GPUs.
pytorch-flows
PyTorch implementations of algorithms for density estimation
sparse-to-dense.pytorch
ICRA 2018 "Sparse-to-Dense: Depth Prediction from Sparse Depth Samples and a Single Image" (PyTorch Implementation)
Synthetic2Realistic
[ECCV 2018]: T2Net: Synthetic-to-Realistic Translation for Depth Estimation Tasks
DrummerNet
Supplementary material of "Deep Unsupervised Drum Transcription", ISMIR 2019
JointVAE_v1
Playing around with JointVAE implementation - WIP