Counting 3,742 Big Data & Machine Learning Frameworks, Toolsets, and Examples...
Suggestion? Feedback? Tweet @stkim1

pytorch_GAN_zoo

475

A mix of GAN toolbox implementations including progressive growing

enas

1069

TensorFlow Code for paper "Efficient Neural Architecture Search via Parameter Sharing"

improved_wgan_training

1556

Code for reproducing experiments in "Improved Training of Wasserstein GANs"

models

51646

Models built with TensorFlow

minigo

2716

An open-source implementation of the AlphaGoZero algorithm

pytorch-faster-rcnn

1130

A pytorch implementation of faster RCNN detection framework based on Xinlei Chen's tf-faster-rcnn.

gcn

2723

Implementation of Graph Convolutional Networks in TensorFlow

gradient-checkpointing

1657

Make huge neural nets fit in memory

flownet2-pytorch

1118

Pytorch implementation of FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks

prototypical-networks

316

Code for the NIPS 2017 Paper "Prototypical Networks for Few-shot Learning"

SampleRNN_torch

116

Torch implementation of SampleRNN: An Unconditional End-to-End Neural Audio Generation Model

faster-rcnn.pytorch

3174

This project is a faster faster R-CNN implementation, aimed to accelerating the training of faster R-CNN object detection models

wav2letter

3554

Facebook AI Research Automatic Speech Recognition Toolkit

StarGAN

3769

StarGAN: Unified Generative Adversarial Networks for Multi-Domain Image-to-Image Translation

pytorch-a2c-ppo-acktr-gail

1277

PyTorch implementation of Advantage Actor Critic (A2C), Proximal Policy Optimization (PPO), Scalable trust-region method for deep reinforcement learning using Kronecker-factored approximation (ACKTR) and Generative Adversarial Imitation Learning (GAIL).

picojs

4603

A face detection library in 200 lines of JavaScript

insightface

3129

Face Recognition Project on MXNet

pytorch-CycleGAN-and-pix2pix

7791

Image-to-image translation in PyTorch (e.g. horse2zebra, edges2cats, and more)

Detectron

20039

Facebook AI Research platform for object detection research, implementing popular algorithms like Mask R-CNN and RetinaNet.

keras-deeplab-v3-plus

717

Keras implementation of Deeplab v3+ with pretrained weights

keras-yolo3

3122

A Keras implementation of YOLOv3 (Tensorflow backend)

YAD2K

1879

Yet Another Darknet 2 Keras

py-faster-rcnn

5641

Faster R-CNN Python implementation

MUNIT-Tensorflow

224

Simple Tensorflow implementation of "Multimodal Unsupervised Image-to-Image Translation"

MUNIT

1466

Multimodal Unsupervised Image-to-Image Translation

SeqGAN

1505

Implementation of Sequence Generative Adversarial Nets with Policy Gradient

poincare-embeddings

1058

PyTorch implementation of the NIPS-17 paper "Poincaré Embeddings for Learning Hierarchical Representations"

deep-head-pose

511

Deep Learning Head Pose Estimation using PyTorch.

pytorch-playground

1433

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

gluon-cv

2270

A Deep Learning Toolkit for Computer Vision