Top 3972 pytorch open source projects

DCGAN LSGAN WGAN-GP DRAGAN PyTorch

A scalable Gensim implementation of "Learning Role-based Graph Embeddings" (IJCAI 2018).

[ICCV 2019] Depth Hints are complementary depth suggestions which improve monocular depth estimation algorithms trained from stereo pairs

Use pytorch to implement word2vec

深度学习与PyTorch入门实战视频教程 配套源代码和PPT

PyTorch implementation of DEC (Deep Embedding Clustering)

The Pytorch Implementation of L-Softmax

Tensor Network Learning with PyTorch

A nnie quantization aware training tool on pytorch.

Pytorch version of Vision Transformer (ViT) with pretrained models. This is part of CASL (https://casl-project.github.io/) and ASYML project.

Pytorch-based tools for visualizing and understanding the neurons of a GAN. https://gandissect.csail.mit.edu/

Deep Learning for LiDAR Point Clouds in Autonomous Driving: A Review

GANimation: Anatomically-aware Facial Animation from a Single Image (ECCV'18 Oral) [PyTorch]

Semantic Segmentation on PyTorch (include FCN, PSPNet, Deeplabv3, Deeplabv3+, DANet, DenseASPP, BiSeNet, EncNet, DUNet, ICNet, ENet, OCNet, CCNet, PSANet, CGNet, ESPNet, LEDNet, DFANet)

Implementation of Convolutional Neural Networks for Sentence Classification (Y.Kim, EMNLP 2014) on Pytorch.

I3D and 3D-ResNets in PyTorch

Implementation of Siamese Networks for image one-shot learning by PyTorch, train and test model on dataset Omniglot

center loss for face recognition

Pytorch implementation of LayoutNet.

Taking a Deeper Look at the Inverse Compositional Algorithm (CVPR 2019, Oral)

Research Code for ICCV 2019 paper "Relation-aware Graph Attention Network for Visual Question Answering"

A PyTorch Implementation of MobileNetv2+DeepLabv3

[EMNLP 2018] PyTorch code for TVQA: Localized, Compositional Video Question Answering

Pytorch port of Google Research's VGGish model used for extracting audio features.

Datasets, networks and transformations for 3D deep learning in PyTorch

Training examples and results for ImageNet(ILSVRC2012)/CIFAR100/COCO2017/VOC2007+VOC2012 datasets.Image Classification/Object Detection.Include ResNet/EfficientNet/VovNet/DarkNet/RegNet/RetinaNet/FCOS/CenterNet/YOLOv3.

[Under development]- Implementation of various methods for dimensionality reduction and spectral clustering implemented with Pytorch

LeagueAI software framework for League of Legends that provides information about the state of the game based on Image Recognition using OpenCV and Pytorch.

Training and inference code for ExpandNet

A PyTorch implementation of OpenAI's REPTILE algorithm

Source codes for the book "Application of Neural Network and PyTorch"

Official code for "Object counting and instance segmentation with image-level supervision", in CVPR 2019 and TPAMI 2020

Exploration of different solutions to action recognition in video, using neural networks implemented in PyTorch.

Efficient Spiking Neural Network framework, built on top of PyTorch for GPU acceleration

Pitfalls of In-Domain Uncertainty Estimation and Ensembling in Deep Learning, ICLR 2020

Pytorch implementation of network design paradigm described in the paper "Designing Network Design Spaces"

An open-source toolbox for action understanding based on PyTorch

Speech commands recognition with PyTorch

jupyter/ipython experiment containers for GPU and general RAM re-use

PyTorch implementation of DCGAN, WGAN-GP and SNGAN.

Official Pytorch Implementation of 'Background Suppression Network for Weakly-supervised Temporal Action Localization' (AAAI-20)

Object detection. EfficientDet-D5 level COCO AP in 20 epochs. SOTA single-stage detector on Waymo Open Dataset.

Pytorch Implementation of recent visual attribution methods for model interpretability

Implementation of CRAFT Text Detection

A lightweight contextual bandit & reinforcement learning library designed to be used in production Python services.

Paper and implementation of UNet-related model.

convolutional lstm implementation in pytorch

PyTorch implementation of Hash Embeddings (NIPS 2017). Submission to the NIPS Implementation Challenge.

A metric for Perceptual Image-Error Assessment through Pairwise Preference (PieAPP at CVPR 2018).

face image illumination quality assessment implement by pytorch

Code for Fast Neural Architecture Search of Compact Semantic Segmentation Models via Auxiliary Cells, CVPR '19

MONeT framework for reducing memory consumption of DNN training

Graph Neural Network Library for PyTorch

a pytorch implementation of auto-punctuation learned character by character

A collection of deep learning models implemented in PyTorch

State Representation Learning (SRL) zoo with PyTorch - Part of S-RL Toolbox

Understanding and visualizing PyTorch Batching with LSTM

This implements training of popular model architectures, such as AlexNet, ResNet and VGG on the ImageNet dataset(Now we supported alexnet, vgg, resnet, squeezenet, densenet)

RefineNet-101 VOC in PyTorch