ml-lab Goto Github PK

pytorch-mobilenet-v3

MobileNetV3 in pytorch and ImageNet pretrained models

pytorch-neucom

Pytorch implementation of DeepMind's differentiable neural computer paper.

pytorch-neural-search-optimizer

PyTorch implementation of Neural Optimizer Search's Optimizer_1

pytorch-noreward-rl

pytorch implementation of Curiosity-driven Exploration by Self-supervised Prediction

pytorch-ntm

A Pytorch implementation of an NTM (Neural Turing Machine)

pytorch-openai-transformer-lm

A PyTorch implementation of OpenAI's finetuned transformer language model with a script to import the weights pre-trained by OpenAI

pytorch-pose

A PyTorch toolkit for 2D Human Pose Estimation.

pytorch-pretrained-bert

PyTorch version of Google AI's BERT model with script to load Google's pre-trained models

pytorch-qrnn

PyTorch implementation of the Quasi-Recurrent Neural Network - up to 16 times faster than NVIDIA's cuDNN LSTM

pytorch-reinforce

PyTorch Implementation of REINFORCE for both discrete & continuous control

pytorch-retinanet

RetinaNet in PyTorch

pytorch-retinanet-1

Pytorch implementation of RetinaNet object detection.

pytorch-rotationnet

pytorch-semseg

Semantic Segmentation Architectures Implemented in PyTorch

pytorch-sepconv

an implementation of Video Frame Interpolation via Adaptive Separable Convolution using PyTorch

pytorch-seq2seq

A framework for sequence-to-sequence (seq2seq) models in PyTorch.

pytorch-siamfc

Pytorch implementation of "Fully-Convolutional Siamese Networks for Object Tracking"

pytorch-siggraph2017-inpainting

You can try inpainting easily.

pytorch-slimming

Learning Efficient Convolutional Networks through Network Slimming, In ICCV 2017.

pytorch-smoothgrad

SmoothGrad implementation in PyTorch

pytorch-struct

Fast, general, and tested differentiable structured prediction in PyTorch

pytorch-studiogan

StudioGAN is a Pytorch library providing the implementation of representative Generative Adversarial Networks (GANs) for conditional/unconditional image synthesis.

pytorch-style-transfer

Neural Style and MSG-Net

pytorch-unet

Pytorch implementation of the U-Net for image semantic segmentation, with dense CRF post-processing

pytorch-video-pipeline

pytorch-voxel-flow

pytorch-wordemb

Load pretrained word embeddings (word2vec, glove format) into torch.FloatTensor for PyTorch

pytorch-yolo2

pytorch yolo2

pytorch-yolov3

Minimal PyTorch implementation of YOLOv3

pytorch.repmet

THIS PROJECT IS BEING REWORKED TO BE MORE READABLE ALSO CURRENTLY ADDING DETECTION