mi18

faster-rcnn.pytorch

A faster pytorch implementation of faster r-cnn

a-PyTorch-Tutorial-to-Image-Captioning

Show, Attend, and Tell | a PyTorch Tutorial to Image Captioning

STAR

[ECCV 2020] Code for "Spatio-Temporal Graph Transformer Networks for Pedestrian Trajectory Prediction"

time_series_forecasting_pytorch

time series forecasting using pytorch，including ANN,RNN,LSTM,GRU and TSR-RNN，experimental code

anticipating-activities

routing-transformer

Fully featured implementation of Routing Transformer

VideoTransformer

TransformerDemo

Pytorch nn.Transformer Demo

VideoVRD-in-pytorch

BERT-pytorch

Google AI 2018 BERT pytorch implementation

VL-BERT

Code for ICLR 2020 paper "VL-BERT: Pre-training of Generic Visual-Linguistic Representations".

charades-algorithms

Activity Recognition Algorithms for the Charades Dataset

video-caption.pytorch

pytorch implementation of video captioning

two-stream-pytorch

PyTorch implementation of two-stream networks for video action recognition

ActionGenome

A video database bridging human actions and human-object relationships

charades-algorithms

Train and Read Charades Dataset, implemented on Python and PyTorch, on Windows

action-recognition

Three steps to train your own model for action recognition based on CNN and LSTM by PyTorch.

superpixel_fcn

[CVPR‘20] SpixelFCN: Superpixel Segmentation with Fully Convolutional Network

Pytorch-UNet

PyTorch implementation of the U-Net for image semantic segmentation with high quality images

unet

unet for image segmentation

Unet-Segmentation-Pytorch-Nest-of-Unets

Implementation of different kinds of Unet Models for Image Segmentation - Unet , RCNN-Unet, Attention Unet, RCNN-Attention Unet, Nested Unet

tensorflow-deeplab-v3

DeepLabv3 built in TensorFlow

SlowFast

PySlowFast: video understanding codebase from FAIR for reproducing state-of-the-art video models.

PRML

PRML algorithms implemented in Python

pytorch_ssn

A pytorch version of SSN

google-research

Google Research

knowledge_graph_attention_network

KGAT: Knowledge Graph Attention Network for Recommendation, KDD2019

GraphSAGE

Representation learning on large graphs using stochastic graph convolutions.

pretrained-models.pytorch

Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.

moments_models

The pretrained models trained on Moments in Time Dataset