ghostxsl

PyTorch-GAN

PyTorch implementations of Generative Adversarial Networks.

detectorch

Detectorch - detectron for PyTorch

RoIAlign.pytorch

RoIAlign & crop_and_resize for PyTorch

Augmentor

Image augmentation library in Python for machine learning.

Detectron.pytorch

A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.

Mask-RCNN

A PyTorch implementation of the architecture of Mask RCNN

pytorch-mask-rcnn

cocoapi

COCO API - Dataset @ http://cocodataset.org/

yolo3_pytorch

Yolo3 Implementation in Pytorch using COCO and BDD100K dataset. Forked from https://github.com/BobLiu20/YOLOv3_PyTorch.git. For my own fun, so the introduction is not accurate.

darknet

YOLOv4 / Scaled-YOLOv4 / YOLO - Neural Networks for Object Detection (Windows and Linux version of Darknet )

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

EasyTrain

12306自动抢票软件

pytorch-rl

Deep Reinforcement Learning with pytorch & visdom

DeepRL

Modularized Implementation of Deep RL Algorithms in PyTorch

gym

A toolkit for developing and comparing reinforcement learning algorithms.

Wechat_AutoJump

AI plays WeChat Jump Game

pytorch-GAN

A minimal implementaion (less than 150 lines of code with visualization) of DCGAN/WGAN in PyTorch with jupyter notebooks

DenseNet

Densely Connected Convolutional Networks, In CVPR 2017 (Best Paper Award).

GetToThePoint

Pytorch implementation of "Get to the point: Get To The Point: Summarization with Pointer-Generator Networks"

pointer-generator

Code for the ACL 2017 paper "Get To The Point: Summarization with Pointer-Generator Networks"

examples

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

pytorch-playground

Base pretrained models and datasets in pytorch (MNIST, SVHN, CIFAR10, CIFAR100, STL10, AlexNet, VGG16, VGG19, ResNet, Inception, SqueezeNet)

tensorflow

An Open Source Machine Learning Framework for Everyone

WassersteinGAN

deep-residual-networks

Deep Residual Learning for Image Recognition

progressive_growing_of_gans

Progressive Growing of GANs for Improved Quality, Stability, and Variation

caffe

Caffe: a fast open framework for deep learning.