Top 3972 pytorch open source projects

A High-Quality PyTorch Implementation of "Globally and Locally Consistent Image Completion".

An easy implementation of Faster R-CNN (https://arxiv.org/pdf/1506.01497.pdf) in PyTorch.

opencv reimplement for transforms in torchvision

How and why you want to make your pytorch CUDA/CPP extension with a Makefile

pytorch implement of Lookahead Optimizer

Task-based end-to-end model learning in stochastic optimization

A python implementation of Deep-Image-Analogy based on pytorch.

Code for EMNLP18 paper "Spherical Latent Spaces for Stable Variational Autoencoders"

👋 ML/DL学习笔记（基础+论文）

Implementation of the Chamfer Distance as a module for pyTorch

Neural architecture search(NAS)

Numenta Platform for Intelligent Computing PyTorch libraries

Official PyTorch implementation of NICE-GAN: Reusing Discriminators for Encoding: Towards Unsupervised Image-to-Image Translation

Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory

An offline deep reinforcement learning library

Complete-Life-Cycle-of-a-Data-Science-Project

Node.js binding for PyTorch.

PyTorch Implementation of Fully Convolutional Networks (a very simple and easy demo).

Helsinki Prosody Corpus and A System for Predicting Prosodic Prominence from Text

A fast, high-quality neural vocoder.

DiffWave is a fast, high-quality neural vocoder and waveform synthesizer.

SOTA Re-identification Methods and Toolbox

This project is an unofficial implementation of AlphaGAN: Generative adversarial networks for natural image matting published at the BMVC 2018

VQ-VAE for Acoustic Unit Discovery and Voice Conversion

Fast and accurate single-person pose estimation, ranked 10th at CVPR'19 LIP challenge. Contains implementation of "Global Context for Convolutional Pose Machines" paper.

Deep-Learning based CTR models implemented by PyTorch

AlphaPose + ST-GCN + SORT.

[ECCV 2020] PSConv: Squeezing Feature Pyramid into One Compact Poly-Scale Convolutional Layer

High-quality Neural Networks for Computer Vision 😎

A PyTorch implementation of Mnemonic Reader for the Machine Comprehension task

Breast density classification with deep convolutional neural networks

Implement some models of RGB/RGBD semantic segmentation in PyTorch, easy to run. Such as FCN, RefineNet, PSPNet, RDFNet, 3DGNN, PointNet, DeepLab V3, DeepLab V3 plus, DenseASPP, FastFCN

Autoregressive Predictive Coding: An unsupervised autoregressive model for speech representation learning

TensorDash is an application that lets you remotely monitor your deep learning model's metrics and notifies you when your model training is completed or crashed.

Official implementation of FQ-GAN

This is Top 3 Code for the Person ReID Compitition of NAIC

NCRF++, a Neural Sequence Labeling Toolkit. Easy use to any sequence labeling tasks (e.g. NER, POS, Segmentation). It includes character LSTM/CNN, word LSTM/CNN and softmax/CRF components.

An IPython Notebook tutorial on deep learning for natural language processing, including structure prediction.

Reproduce CheXNet

[ACM MM 2018 Oral] GestureGAN for Hand Gesture-to-Gesture Translation in the Wild

Code for 3rd place solution in Kaggle Humpback Whale Identification Challenge.

Spatial Attentive Single-Image Deraining with a High Quality Real Rain Dataset (CVPR'19)

PyTorch 101 series covering everything from the basic building blocks all the way to building custom architectures.

BERT for Finance : UC Berkeley MIDS w266 Final Project

pytorch implementation for Patient Knowledge Distillation for BERT Model Compression

Attention-guided CNN for image denoising(Neural Networks,2020)

Allosaurus is a pretrained universal phone recognizer for more than 2000 languages

A clean and simple data loading library for Continual Learning

The 3rd Place Submission to AICity Challenge 2020 Track2

Official code for Cotatron @ INTERSPEECH 2020

[WACV2021] Foreground-aware Semantic Representations for Image Harmonization https://arxiv.org/abs/2006.00809

Official Pytorch Code of KiU-Net for Image Segmentation - MICCAI 2020 (Oral)

implement infoGAN using pytorch

Ready-to-use OCR with 80+ supported languages and all popular writing scripts including Latin, Chinese, Arabic, Devanagari, Cyrillic and etc.

Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.

High-Level Training, Data Augmentation, and Utilities for Pytorch

A PyTorch implementation of MPIIGaze and MPIIFaceGaze

Pytorch implementation of "One-Shot Unsupervised Cross Domain Translation" NIPS 2018

The official PyTorch implementation of the paper "Learning by Analogy: Reliable Supervision from Transformations for Unsupervised Optical Flow Estimation".

Pytorch Imagenet Models Example + Transfer Learning (and fine-tuning)