Stars
🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.
Command-line program to download videos from YouTube.com and other video sites
PyTorch Tutorial for Deep Learning Researchers
Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow
Image-to-Image Translation in PyTorch
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
PyTorch implementations of Generative Adversarial Networks.
OpenAI Baselines: high-quality implementations of reinforcement learning algorithms
ChatterBot is a machine learning, conversational dialog engine for creating chat bots
Refine high-quality datasets and visual AI models
A PyTorch implementation of the Transformer model in "Attention is All You Need".
Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.
tensorboard for pytorch (and chainer, mxnet, numpy, ...)
A faster pytorch implementation of faster r-cnn
BertViz: Visualize Attention in NLP Models (BERT, GPT2, BART, etc.)
Collection of generative models, e.g. GAN, VAE in Pytorch and Tensorflow.
Open Source Neural Machine Translation and (Large) Language Models in PyTorch
NeuralTalk is a Python+numpy project for learning Multimodal Recurrent Neural Networks that describe images with sentences.
A PyTorch Implementation of Single Shot MultiBox Detector
Most popular metrics used to evaluate object detection algorithms.
Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset
A python wrapper for Alpha Vantage API for financial data.
Python Socket.IO server and client
LPIPS metric. pip install lpips
Deformable Convolutional Networks