pytorch-flows
by ikostrikov
PyTorch implementations of algorithms for density estimation
AI summary
Density estimator library
PyTorch implementations of algorithms for density estimation using invertible transformations.
- stars
- 577
- forks
- 75
- watching
- 18
Similar projects
Found by comparing what the projects do, not just their names.
Autoencoder density estimator
An implementation of a masked autoencoder density estimation model in PyTorch
Optimization algorithm
A PyTorch implementation of an optimization algorithm for continuous control and reinforcement learning tasks
Meta learner
A PyTorch implementation of meta-learning using gradient descent to adapt to new tasks.
Flow estimator
PyTorch implementation of a neural network architecture for estimating the optical flow between frames in videos
Optical flow model
An implementation of unsupervised learning for multi-frame optical flow with occlusions using PyTorch.
I3D weight transfer
Transfers pre-trained I3D network weights from TensorFlow to PyTorch
Computer Vision Toolkit
A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision.
Dense reconstruction model
An implementation of CloudWalk's DenseBody paper using PyTorch
Flow library
A PyTorch implementation of volume-preserving flows for variational autoencoders.
Image classifier
A PyTorch project for comparing image classification models and facilitating quick experiment setup
EfficientNet model
A PyTorch implementation of EfficientNet for computer vision tasks
Object descriptor network
This project presents a method for learning dense object descriptors from self-supervised data, enabling tasks such as robotic manipulation and object recognition.
Deep learning toolkit
A Python framework for building deep learning models with optimized encoding layers and batch normalization.
RL algorithm
An implementation of reinforcement learning algorithms for continuous control tasks using deep neural networks.
Pose estimator
A PyTorch implementation of a deep learning model for 3D multi-person pose estimation from a single RGB image