qiqihaer / Awsome-PyTorch-Repos-and-Datasets

Awsome PyTorch Repos and Datasets

Awsome PyTorch Repos and Datasets

Email: [e-mail]

• Awsome PyTorch Repos and Datasets
  • PyTorch Repository
    • NLP
    • Reinforcement Learning
    • Semantic Segmentation
    • Object Detection
    • Pose Estimation
    • GAN
    • Research
    • Course
    • Others
  • Datasets
    • Computer Vision
    • Medical

PyTorch Repository

NLP

1. allennlp An open-source NLP research library, built on PyTorch.
2. pytext PyText is a deep-learning based NLP modeling framework built on PyTorch. PyText addresses the often-conflicting requirements of enabling rapid experimentation and of serving models at scale.
3. text Data loaders and abstractions for text and NLP.
4. fairseq Fairseq(-py) is a sequence modeling toolkit that allows researchers and developers to train custom models for translation, summarization, language modeling and other text generation tasks. It provides reference implementations of various sequence-to-sequence models.

Reinforcement Learning

1. Deep-Reinforcement-Learning-Algorithms-with-PyTorch The repository contains PyTorch implementations of deep reinforcement learning algorithms.
2. Horizon A platform for Applied Reinforcement Learning (Applied RL).

Semantic Segmentation

1. pytorch-semantic-segmentation This repository contains some models for semantic segmentation and the pipeline of training and testing models, implemented in PyTorch.

Object Detection

1. mmdetection mmdetection is an open source object detection toolbox based on PyTorch.
2. RCF-pytorch Richer Convolutional Features for Edge Detection model in pytorch.
3. maskrcnn-benchmark Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.
4. yolov3 The repo contains inference and training code for YOLOv3 in PyTorch. Training is done on the COCO dataset by default.
5. PyTorch-YOLOv3 Minimal PyTorch implementation of YOLOv3.

Pose Estimation

1. pytorch_Realtime_Multi-Person_Pose_Estimation This is a pytorch version of Realtime_Multi-Person_Pose_Estimation。

GAN

1. torchgan Research Framework for easy and efficient training of GANs based on Pytorch.
2. portrain-gan torch code to decode (and almost encode) latents from art-DCGAN's Portrait GAN.

Research

1. pytorch-3d-point-cloud-generation Pytorch code to construct a 3D point cloud model from single RGB image.
2. 3D-ResNets-PyTorch 3D ResNets for Action Recognition (CVPR 2018).
3. Super-SloMo PyTorch implementation of Super SloMo by Jiang et al.
4. uis-rnn This is the library for the Unbounded Interleaved-State Recurrent Neural Network (UIS-RNN) algorithm, corresponding to the paper Fully Supervised Speaker Diarization.
5. pytorch-pretrained-BERT A PyTorch implementation of Google AI's BERT model provided with Google's pre-trained models, examples and utilities.
6. rethinking-network-pruning Rethinking the Value of Network Pruning (Pytorch) (ICLR 2019).
7. BERT-pytorch Google AI 2018 BERT pytorch implementation.
8. CornerNet
9. pnn.pytorch PyTorch implementation of CVPR'18 - Perturbative Neural Networks http://xujuefei.com/pnn.html.
10. loss-landscape Code for visualizing the loss landscape of neural nets.

Course

1. fastai The fastai deep learning library, plus lessons and and tutorials.
2. course-v3 The 3rd edition of course.fast.ai - coming in 2019.

Others

1. apex A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch.
2. ignite High-level library to help with training neural networks in PyTorch.

Datasets

Computer Vision

1. tencent-ml-images Largest multi-label image database.

Medical

1. fastMRI A large-scale dataset of both raw MRI measurements and clinical MRI images.