Convolution_LSTM_pytorch: A multi-layer convolution LSTM module
face-alignment:
?
2D and 3D Face alignment library build using pytorch adrianbulat.com
pytorch-semantic-segmentation: PyTorch for Semantic Segmentation.
RoIAlign.pytorch: This is a PyTorch version of RoIAlign. This implementation is based on crop_and_resize and supports both forward and backward on CPU and GPU.
pytorch-cnn-finetune: Fine-tune pretrained Convolutional Neural Networks with PyTorch.
detectorch: Detectorch - detectron for PyTorch
Augmentor: Image augmentation library in Python for machine learning. http://augmentor.readthedocs.io
s2cnn: This library contains a PyTorch implementation of the SO(3) equivariant CNNs for spherical signals (e.g. omnidirectional cameras, signals on the globe) s
三. 概率生成库
ptstat: Probabilistic Programming and Statistical Inference in PyTorch
pyro: Deep universal probabilistic programming with Python and PyTorch http://pyro.ai
probtorch: Probabilistic Torch is library for deep generative models that extends PyTorch.
paysage: Unsupervised learning and generative models in python/pytorch.
pyvarinf: Python package facilitating the use of Bayesian Deep Learning methods with Variational Inference for PyTorch.
pyprob: A PyTorch-based library for probabilistic programming and inference compilation.
mia: A library for running membership inference attacks against ML models.
四. 其他库
pytorch extras : Some extra features for pytorch.
functional zoo : PyTorch, unlike lua torch, has autograd in it's core, so using modular structure of torch.nn modules is not necessary, one can easily allocate needed Variables and write a function that utilizes them, which is sometimes more convenient. This repo contains model definitions in this functional way, with pretrained weights for some models.
torch-sampling : This package provides a set of transforms and data structures for sampling from in-memory or out-of-memory data.
torchcraft-py : Python wrapper for TorchCraft, a bridge between Torch and StarCraft for AI research.
aorun : Aorun intend to be a Keras with PyTorch as backend.
logger : A simple logger for experiments.
PyTorch-docset : PyTorch docset! use with Dash, Zeal, Velocity, or LovelyDocs.
convert_torch_to_pytorch : Convert torch t7 model to pytorch model and source.
pretrained-models.pytorch: The goal of this repo is to help to reproduce research papers results.
pytorch_fft : PyTorch wrapper for FFTs
五. 教程例子
Practical Pytorch : Tutorials explaining different RNN models
DeepLearningForNLPInPytorch : An IPython Notebook tutorial on deep learning, with an emphasis on Natural Language Processing.
pytorch-tutorial : tutorial for researchers to learn deep learning with pytorch.
pytorch-exercises : pytorch-exercises collection.
pytorch tutorials : Various pytorch tutorials.
pytorch examples : A repository showcasing examples of using pytorch
pytorch practice : Some example scripts on pytorch.
pytorch mini tutorials : Minimal tutorials for PyTorch adapted from Alec Radford's Theano tutorials.
pytorch text classification : A simple implementation of CNN based text classification in Pytorch
cats vs dogs : Example of network fine-tuning in pytorch for the kaggle competition Dogs vs. Cats Redux: Kernels Edition. Currently #27 (0.05074) on the leaderboard.
六. 论文实现
google_evolution : This implements one of result networks from Large-scale evolution of image classifiers by Esteban Real, et. al.
pyscatwave : Fast Scattering Transform with CuPy/PyTorch,read the paper here
scalingscattering : Scaling The Scattering Transform : Deep Hybrid Networks.
deep-auto-punctuation : a pytorch implementation of auto-punctuation learned character by character.
Realtime_Multi-Person_Pose_Estimation : This is a pytorch version of Realtime_Multi-Person_Pose_Estimation, origin code is here .
PyTorch-value-iteration-networks : PyTorch implementation of the Value Iteration Networks (NIPS '16) paper
pytorch_Highway : Highway network implemented in pytorch.
pytorch_NEG_loss : NEG loss implemented in pytorch.
pytorch_RVAE : Recurrent Variational Autoencoder that generates sequential data implemented in pytorch.
pytorch_TDNN : Time Delayed NN implemented in pytorch.
七. 其他Pytorch资源
the-incredible-pytorch : The Incredible PyTorch: a curated list of tutorials, papers, projects, communities and more relating to PyTorch.
generative models : Collection of generative models, e.g. GAN, VAE in Tensorflow, Keras, and Pytorch. http://wiseodd.github.io
pytorch vs tensorflow : an informative thread on reddit.
Pytorch discussion forum
pytorch notebook: docker-stack : A project similar to Jupyter Notebook Scientific Python Stack
drawlikebobross: Draw like Bob Ross using the power of Neural Networks (With PyTorch)!
pytorch-tvmisc: Totally Versatile Miscellanea for Pytorch
pytorch-a3c-mujoco: Implement A3C for Mujoco gym envs.
PyTorch in 5 Minutes.
pytorch_chatbot: A Marvelous ChatBot implemented using PyTorch.