PyTorch implementation of "WILDCAT: Weakly Supervised Learning of Deep ConvNets for Image Classification, Pointwise Localization and Segmentation", CVPR 2017

✭ 238

python deep-learning pytorch image-classification cvpr-2017

Chexnet With Localization

Weakly Supervised Learning for Findings Detection in Medical Images

✭ 238

python deep-learning pytorch medical-imaging densenet bounding-boxes

Enet Real Time Semantic Segmentation

ENet - A Neural Net Architecture for real time Semantic Segmentation

✭ 238

python3 jupyter-notebook deep-learning pytorch artificial-intelligence paper

Angular Penalty Softmax Losses Pytorch

Angular penalty loss functions in Pytorch (ArcFace, SphereFace, Additive Margin, CosFace)

✭ 236

python pytorch face-recognition metric-learning

Pytorch Bert Crf Ner

KoBERT와 CRF로 만든 한국어 개체명인식기 (BERT+CRF based Named Entity Recognition model for Korean)

✭ 236

jupyter-notebook pytorch nlp natural-language-processing named-entity-recognition ner korean crf

Lhotse

✭ 236

python deep-learning machine-learning pytorch audio ai data speech kaldi

Tianshou

An elegant PyTorch deep reinforcement learning library.

✭ 4,109

python Makefile pytorch dqn ppo policy-gradient imitation-learning ddpg mujoco benchmark library rl cql atari sac drl npg double-dqn trpo a2c td3 bcq

Deepnlp Models Pytorch

Pytorch implementations of various Deep NLP models in cs-224n(Stanford Univ)

✭ 2,760

Jupyter Notebook deep-learning pytorch nlp neural-network natural-language-processing rnn deep-nlp-models stanford-univ cs-224n

Pytorch Vae

A Variational Autoencoder (VAE) implemented in PyTorch

✭ 237

python machine-learning pytorch autoencoder pytorch-tutorials

Inferno

A utility library around PyTorch

✭ 237

python deep-learning pytorch neural-networks

University1652 Baseline

ACM Multimedia2020 University-1652: A Multi-view Multi-source Benchmark for Drone-based Geo-localization 🚁 annotates 1652 buildings in 72 universities around the world.

✭ 232

python pytorch awesome-list dataset apex image-retrieval drone satellite

Supergo

A student implementation of Alpha Go Zero

✭ 236

python python3 machine-learning pytorch reinforcement-learning

Relational Rnn Pytorch

An implementation of DeepMind's Relational Recurrent Neural Networks in PyTorch.

✭ 236

python deep-learning pytorch transformer recurrent-neural-networks language-model deepmind

Flambe

An ML framework to accelerate research and its path to production.

✭ 236

python deep-learning machine-learning pytorch distributed research ml

Fcos

FCOS: Fully Convolutional One-Stage Object Detection (ICCV'19)

✭ 2,839

python Cuda C++Dockerfile pytorch computer-vision object-detection one-stage fcos anchor-free iccv2019

Dlwpt Code

Code for the book Deep Learning with PyTorch by Eli Stevens, Luca Antiga, and Thomas Viehmann.

✭ 3,054

python Jupyter Notebook deep-learning pytorch deep-neural-networks

Torchnlp

Easy to use NLP library built on PyTorch and TorchText

✭ 233

python machine-learning pytorch nlp transformer crf

Crnn Audio Classification

UrbanSound classification using Convolutional Recurrent Networks in PyTorch

✭ 235

python pytorch audio lstm rnn crnn

Piq

Measures and metrics for image2image tasks. PyTorch.

✭ 230

python python3 pytorch metrics gan

Pyramidnet Pytorch

A PyTorch implementation for PyramidNets (Deep Pyramidal Residual Networks, https://arxiv.org/abs/1610.02915)

✭ 234

python pytorch resnet imagenet

Appnp

A PyTorch implementation of "Predict then Propagate: Graph Neural Networks meet Personalized PageRank" (ICLR 2019).

✭ 234

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

Lcnn

LCNN: End-to-End Wireframe Parsing

✭ 234

python deep-learning pytorch deep-neural-networks cnn line

Alphaction

Spatio-Temporal Action Localization System

✭ 221

python pytorch gpu torch action-recognition ava

Pytorch Drl

PyTorch implementations of various Deep Reinforcement Learning (DRL) algorithms for both single agent and multi-agent.

✭ 233

python pytorch reinforcement-learning deep-reinforcement-learning dqn ppo rl actor-critic ddpg deep-q-network

Ssd Variants

PyTorch implementation of several SSD based object detection algorithms.

✭ 233

python deep-learning pytorch object-detection ssd convolutional-networks

Pytorch Multi Label Classifier

A pytorch implemented classifier for Multiple-Label classification

✭ 232

python pytorch classifier

Meshed Memory Transformer

Meshed-Memory Transformer for Image Captioning. CVPR 2020

✭ 230

python pytorch transformer image-captioning

Retinaface

The remake of the https://github.com/biubug6/Pytorch_Retinaface

✭ 226

python pytorch object-detection deeplearning face-detection

Super Slomo

PyTorch implementation of Super SloMo by Jiang et al.

✭ 2,714

python Jupyter Notebook Dockerfile deep-learning pytorch deep-neural-networks convolutional-neural-networks slow-motion pytorch-implmention frame-interpolation video-frame-interpolation super-slomo slomo

Nanodet

⚡Super fast and lightweight anchor-free object detection model. 🔥Only 980 KB(int8) / 1.8MB (fp16) and run 97FPS on cellphone🔥

✭ 3,640

python C++java CMake android deep-learning pytorch object-detection deep-neural-networks model-zoo mnn shufflenet ncnn openvino anchor-free efficientnet repvgg

🧑‍🏫 50! Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠

✭ 5,720

Jupyter Notebook python deep-learning pytorch reinforcement-learning neural-networks gan deep-learning-tutorial machine-learning transformers transformer optimizers

Person reid baseline pytorch

Pytorch ReID: A tiny, friendly, strong pytorch implement of object re-identification baseline. Tutorial 👉https://github.com/layumi/Person_reID_baseline_pytorch/tree/master/tutorial

Echotorch

A Python toolkit for Reservoir Computing and Echo State Network experimentation based on pyTorch. EchoTorch is the only Python module available to easily create Deep Reservoir Computing models.

✭ 231

python machine-learning pytorch artificial-intelligence neural-networks machine-learning-algorithms recurrent-neural-networks machinelearning torch artificial-neural-networks

Wassdistance

Approximating Wasserstein distances with PyTorch

✭ 229

jupyter-notebook deep-learning machine-learning pytorch

Siren

PyTorch implementation of Sinusodial Representation networks (SIREN)

✭ 232

python deep-learning machine-learning pytorch

Pyconv

Pyramidal Convolution: Rethinking Convolutional Neural Networks for Visual Recognition (https://arxiv.org/pdf/2006.11538.pdf)

✭ 231

python deep-learning machine-learning pytorch computer-vision neural-network artificial-intelligence deep-neural-networks convolutional-neural-networks cnn imagenet image-recognition

Metric Learning Divide And Conquer

Source code for the paper "Divide and Conquer the Embedding Space for Metric Learning", CVPR 2019

✭ 231

python pytorch metric-learning

Hamiltonian Nn

Code for our paper "Hamiltonian Neural Networks"

✭ 229

jupyter-notebook deep-learning pytorch neural-network research physics dynamics

Multigraph transformer

transformer, multi-graph transformer, graph, graph classification, sketch recognition, sketch classification, free-hand sketch, official code of the paper "Multi-Graph Transformer for Free-Hand Sketch Recognition"

✭ 231

python pytorch graph sketch transformer

Jetson Containers

Machine Learning Containers for NVIDIA Jetson and JetPack-L4T

✭ 223

shell machine-learning docker pytorch tensorflow containers numpy pandas scikit-learn nvidia dockerfiles

Tengine

Tengine is a lite, high performance, modular inference engine for embedded device

✭ 4,012