All Categories → Machine Learning → pytorch

Top 3972 pytorch open source projects

OpenMMLab Detection Toolbox and Benchmark

✭ 17,646

python shell Dockerfile pytorch object-detection instance-segmentation ssd faster-rcnn fast-rcnn rpn mask-rcnn retinanet cascade-rcnn

Pytorch Handbook

pytorch handbook是一本开源的书籍，目标是帮助那些希望和使用PyTorch进行深度学习开发和研究的朋友快速入门，其中包含的Pytorch教程全部通过测试保证可以成功运行

✭ 15,817

Jupyter Notebook deep-learning machine-learning pytorch neural-network pytorch-tutorials pytorch-handbook

Simple Implementation of many GAN models with PyTorch.

✭ 227

jupyter-notebook pytorch gan mnist dcgan wgan celeba wgan-gp

A PyTorch implementation of "Graph Classification Using Structural Attention" (KDD 2018).

✭ 227

python machine-learning pytorch reinforcement-learning neural-networks deep-reinforcement-learning transfer-learning attention sklearn

Gradient based Hyperparameter Tuning library in PyTorch

✭ 226

python deep-learning machine-learning pytorch neural-networks automl hyperparameter-tuning

PyTorch dataset extended with map, cache etc. (tensorflow.data like)

✭ 226

python pytorch tensorflow library dataset cache map filter torch disk

Deeprl Grounding

Train an RL agent to execute natural language instructions in a 3D Environment (PyTorch)

✭ 226

python pytorch reinforcement-learning

Source separation

Deep learning based speech source separation using Pytorch

✭ 226

jupyter-notebook deep-learning pytorch audio speech

DeepPruner: Learning Efficient Stereo Matching via Differentiable PatchMatch (ICCV 2019)

✭ 226

python pytorch real-time

Pytorch memonger

Experimental ground for optimizing memory of pytorch models

✭ 223

python pytorch memory-management

A continuous action space version of A3C LSTM in pytorch plus A3G design

✭ 223

python pytorch openai-gym a3c

Minimalist implementation of VQ-VAE in Pytorch

✭ 224

python deep-learning pytorch deep-neural-networks mnist vae

Tunable U-Net implementation in PyTorch

✭ 224

python pytorch deep-neural-networks segmentation unet

Person Attribute Recognition Marketduke

A simple baseline implemented in PyTorch for pedestrian attribute recognition task, evaluating on Market-1501 and DukeMTMC-reID dataset.

✭ 224

python pytorch computer-vision cnn

Fewshotlearning

Pytorch implementation of the paper "Optimization as a Model for Few-Shot Learning"

✭ 223

python pytorch optimization optimization-algorithms

My Awesome Ai Bookmarks

Curated list of my reads, implementations and core concepts of Artificial Intelligence, Deep Learning, Machine Learning by best folk in the world.

✭ 223

python deep-learning machine-learning pytorch tensorflow awesome awesome-list neural-network artificial-intelligence algorithms list deeplearning datascience blogs

Deep recommender models using PyTorch.

✭ 2,623

python shell deep-learning machine-learning pytorch recommender-system matrix-factorization learning-to-rank

Repulsion loss ssd

Repulsion Loss: Detecting Pedestrians in a Crowd. https://arxiv.org/abs/1711.07752

✭ 222

python pytorch ssd

Robust realtime face and facial landmark tracking on CPU with Unity integration

✭ 216

python csharp pytorch unity unity3d face-detection udp cpu tracker onnx face-tracking

Deep Learning / NLP tutorial for Chatbot Developers

✭ 221

jupyter-notebook deep-learning pytorch nlp tutorial chatbot

DeepInf: Social Influence Prediction with Deep Learning

✭ 221

python pytorch social-network

Unit Testing for pytorch, based on mltest

✭ 222

python machine-learning pytorch testing

PyTorch implementation of Deep Ordinal Regression Network for Monocular Depth Estimation

✭ 220

Pytorch cifar10

Pretrained TorchVision models on CIFAR10 dataset (with weights)

✭ 219

python pytorch computer-vision deeplearning pretrained-models cifar10

Timeseries fastai

fastai V2 implementation of Timeseries classification papers.

✭ 221

jupyter-notebook pytorch classification timeseries

multi_task_NLP is a utility toolkit enabling NLP developers to easily train and infer a single model for multiple tasks.

✭ 221

python pytorch nlp named-entity-recognition nlp-library sequence-labeling

NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch

✭ 215

python pytorch paper image-classification deepmind

Tools for PyTorch

✭ 220

python deep-learning machine-learning pytorch deep-neural-networks

Deform conv pytorch

PyTorch Implementation of Deformable Convolution

✭ 217

jupyter-notebook pytorch cnn

Visualisation tool for CNNs in pytorch

✭ 219

python jupyter-notebook deep-learning pytorch neural-network

Lightweight framework for fast prototyping and training deep neural networks with PyTorch and TensorFlow

✭ 220

python deep-learning machine-learning pytorch tensorflow medical-imaging

MoGA: Searching Beyond MobileNetV3

✭ 220

python pytorch imagenet

Pytorch Deep Learning Template

A Pytorch Computer Vision template to quick start your next project! 🚀🚀

✭ 220

python jupyter-notebook deep-learning pytorch computer-vision

Softmax Splatting

an implementation of softmax splatting for differentiable forward warping using PyTorch

✭ 218

python deep-learning pytorch cuda cupy

Retrieval 2017 Cam

Class-Weighted Convolutional Features for Image Retrieval (BMVC 2017)

✭ 219

python deep-learning pytorch keras convolutional-neural-networks transfer-learning image-retrieval vgg16 cam

Temporal Causal Discovery Framework (PyTorch): discovering causal relationships between time series

✭ 217

python jupyter-notebook machine-learning pytorch neural-networks cnn timeseries prediction forecasting

Lightnetplusplus

LightNet++: Boosted Light-weighted Networks for Real-time Semantic Segmentation

✭ 218

python pytorch semantic-segmentation attention-mechanism apex mobilenetv2 cityscapes

[ICCV 2019 (Oral)] Temporal Attentive Alignment for Large-Scale Video Domain Adaptation (PyTorch)

✭ 217

python pytorch video action-recognition domain-adaptation

How To Read Pytorch

Quick, visual, principled introduction to pytorch code through five colab notebooks.

✭ 218

jupyter-notebook pytorch tutorial

Magnetloss Pytorch

PyTorch implementation of a deep metric learning technique called "Magnet Loss" from Facebook AI Research (FAIR) in ICLR 2016.

✭ 217

python deep-learning docker pytorch cnn classification embeddings metric-learning anaconda

Synchronized Multi-GPU Batch Normalization

✭ 217

python deep-learning pytorch

PyTorch 3D video classification models pre-trained on 65 million Instagram videos

✭ 217

python machine-learning pytorch action-recognition

PyTorch 1.0 官方文档中文版，欢迎关注微信公众号：磐创AI

✭ 215

pytorch pytorch-tutorial

PyTorch implementation of "Efficient Neural Architecture Search via Parameters Sharing"

✭ 2,506

python shell pytorch neural-architecture-search google-brain

A PyTorch Library for Accelerating 3D Deep Learning Research

✭ 2,794

python Cuda C++javascript cython HTML pytorch artificial-intelligence neural-networks model-zoo 3d-deep-learning differentiable-rendering

Python library assists deep learning on graphs

✭ 215

python deep-learning pytorch

EMNLP 2020: "Dialogue Response Ranking Training with Large-Scale Human Feedback Data"

✭ 216

python pytorch dataset dialog pretrained-models

Pytorch implementation for LapSRN (CVPR2017)

✭ 215

python pytorch deep-neural-networks super-resolution cvpr-2017

Fast portrait segmentation

Fast (aimed to "real time") Portrait Segmentation on mobile phone

✭ 215

python pytorch real-time segmentation

BERT Dureader多文档阅读理解排名第七

✭ 215

python pytorch nlp transfer-learning qa

PyTorch implementation of Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning

✭ 213

jupyter-notebook deep-learning pytorch unsupervised-learning representation-learning

Stereo Transformer

Official Repo for Stereo Transformer: Revisiting Stereo Depth Estimation From a Sequence-to-Sequence Perspective with Transformers.

✭ 211

jupyter-notebook pytorch

The PyTorch Implementation of SummaRuNNer

✭ 214

python pytorch summary

Deepfake Detection

Towards deepfake detection that actually works

✭ 213

python deep-learning machine-learning pytorch transfer-learning resnet

Haste: a fast, simple, and open RNN library

✭ 214

python cpp deep-learning machine-learning pytorch tensorflow api algorithm cuda lstm rnn gru

Mot neural solver

Official PyTorch implementation of "Learning a Neural Solver for Multiple Object Tracking" (CVPR 2020 Oral).

✭ 213

Ttnet Real Time Analysis System For Table Tennis Pytorch

Unofficial implementation of "TTNet: Real-time temporal and spatial video analysis of table tennis" (CVPR 2020)

✭ 211

python pytorch segmentation

Deepinfomaxpytorch

Learning deep representations by mutual information estimation and maximization

✭ 212

python deep-learning pytorch compression autoencoder

ICCV 2019 (oral) RankSRGAN: Generative Adversarial Networks with Ranker for Image Super-Resolution. PyTorch implementation

✭ 213

python pytorch gan generative-adversarial-network super-resolution

121-180 of 3972 pytorch projects