PpnpPPNP & APPNP models from "Predict then Propagate: Graph Neural Networks meet Personalized PageRank" (ICLR 2019)
SplitterA Pytorch implementation of "Splitter: Learning Node Representations that Capture Multiple Social Contexts" (WWW 2019).
VipVideo Platform for Action Recognition and Object Detection in Pytorch
TorchfuncPyTorch functions and utilities to make your life easier
Vision3dResearch platform for 3D object detection in PyTorch.
Xinggan[ECCV 2020] XingGAN for Person Image Generation
ImgclsmobSandbox for training deep learning networks
MarsMars is a tensor-based unified framework for large-scale data computation which scales numpy, pandas, scikit-learn and Python functions.
Total3dunderstandingImplementation of CVPR'20 Oral: Total3DUnderstanding: Joint Layout, Object Pose and Mesh Reconstruction for Indoor Scenes from a Single Image
Gst Tacotron A PyTorch implementation of Style Tokens: Unsupervised Style Modeling, Control and Transfer in End-to-End Speech Synthesis
QmlIntroductions to key concepts in quantum machine learning, as well as tutorials and implementations from cutting-edge QML research.
Solov2SOLOv2: Dynamic, Faster and Stronger, achives 39.5mAP on coco test-dev (36 epochs result)
Pytorch PlaygroundBase pretrained models and datasets in pytorch (MNIST, SVHN, CIFAR10, CIFAR100, STL10, AlexNet, VGG16, VGG19, ResNet, Inception, SqueezeNet)
Kd libA Pytorch Knowledge Distillation library for benchmarking and extending works in the domains of Knowledge Distillation, Pruning, and Quantization.
AdeptrlReinforcement learning framework to accelerate research
Pytorch sacPyTorch implementation of Soft Actor-Critic (SAC)
Machine Learning Is All You Need🔥🌟《Machine Learning 格物志》: ML + DL + RL basic codes and notes by sklearn, PyTorch, TensorFlow, Keras & the most important, from scratch!💪 This repository is ALL You Need!
Pyhfpure-Python HistFactory implementation with tensors and autodiff
Torch DctDCT (discrete cosine transform) functions for pytorch
TorchsparseA high-performance neural network library for point cloud processing.
PaiResource scheduling and cluster management for AI
Seal CiA PyTorch implementation of "Semi-Supervised Graph Classification: A Hierarchical Graph Perspective" (WWW 2019)
Nsga NetNSGA-Net, a Neural Architecture Search Algorithm
DiffaiA certifiable defense against adversarial examples by training neural networks to be provably robust
Clipbert[CVPR 2021 Oral] Official PyTorch code for ClipBERT, an efficient framework for end-to-end learning for image-text and video-text tasks.
AutoclintA specially designed light version of Fast AutoAugment
Edge ConnectEdgeConnect: Structure Guided Image Inpainting using Edge Prediction, ICCV 2019 https://arxiv.org/abs/1901.00212
3d BoundingboxPyTorch implementation for 3D Bounding Box Estimation Using Deep Learning and Geometry
PysurvivalOpen source package for Survival Analysis modeling
Pytorch Grad CamMany Class Activation Map methods implemented in Pytorch for CNNs and Vision Transformers. Including Grad-CAM, Grad-CAM++, Score-CAM, Ablation-CAM and XGrad-CAM
Pytorch Kaldipytorch-kaldi is a project for developing state-of-the-art DNN/RNN hybrid speech recognition systems. The DNN part is managed by pytorch, while feature extraction, label computation, and decoding are performed with the kaldi toolkit.
GannotationGANnotation (PyTorch): Landmark-guided face to face synthesis using GANs (And a triple consistency loss!)