Deep Learning

Frameworks for Neural Networks and Deep Learning.

Newest releases

bcmi A curated list of papers, code and resources pertaining to image composition, including image blending, image harmonization, shadow generation, and object placement.

PRBonn Moving Object Segmentation in 3D LiDAR Data: A Learning-based Approach Exploiting Sequential Data (RAL/IROS 2021)

xinntao Real-ESRGAN aims at developing Practical Algorithms for General Image Restoration.

ZhendongWang6 An official repository for Paper "Uformer: A General U-Shaped Transformer for Image Restoration".

kemaloksuz Official PyTorch Implementation of Rank & Sort Loss [ICCV2021]

isaaccorley PyTorch implementation of popular datasets and models in remote sensing tasks (Change Detection, Image Super Resolution, Land Cover Classification/Segmentation, Image-to-Image Translation, etc.) for various Optical (Sentinel-2, La

Jeff-sjtu Code for "Human Pose Regression with Residual Log-likelihood Estimation", ICCV 2021 Oral

naver-ai This code is official implementation of "NeuralWOZ: Learning to Collect Task-Oriented Dialogue via Model-based Simulation".

ShoufaChen Implementation of "A MLP-like Architecture for Dense Prediction"

ToyotaResearchInstitute Code release for our paper, "SimNet: Enabling Robust Unknown Object Manipulation from Pure Synthetic Data via Stereo"

Junjue-Wang FactSeg: Foreground Activation Driven Small Object Semantic Segmentation in Large-Scale Remote Sensing Imagery (TGRS)

facebookresearch Reference implementation of code generation projects from Facebook AI Research. General toolkit to apply machine learning to code, from dataset creation to model training and evaluation. Comes with pretrained models.

sokrypton Making Protein folding accessible to all via Google Colab!

lucidrains Implementation of Invariant Point Attention as a standalone module, which was used in the structure module of Alphafold2 for coordinate refinement.

facebookresearch DrQ-v2: Improved Data-Augmented Reinforcement Learning

sicxu Accurate 3D Face Reconstruction with Weakly-Supervised Learning: From Single Image to Image Set (CVPRW 2019). A PyTorch implementation

Megvii-BaseDetection YOLOX is an anchor-free version of YOLO, with a simpler design but better performance! It aims to bridge the gap between research and industrial communities.

mohamedhassanmus Populating 3D Scenes by Learning Human-Scene Interaction

ShopRunner Collie is a library for preparing, training, and evaluating implicit deep learning hybrid recommender systems, named after the Border Collie dog breed.

tusharsarkar3 XBNET that is built on PyTorch combines tree-based models with neural networks to create a robust architecture that is trained by using a novel optimization technique, Boosted Gradient Descent for Tabular Data which increases its

Jingjing-NLP VOLT(-py) is a vocabulary learning codebase that allows researchers and developers to automaticaly generate a vocabulary with suitable granularity for machine translation.

Suvoo Implementation of Image Style Transfer Using CNNs using Pytorch.

robertmartin8 Using python and scikit-learn to make stock predictions

keonlee9420 PyTorch Implementation of VAENAR-TTS: Variational Auto-Encoder based Non-AutoRegressive Text-to-Speech Synthesis.

TaoRuijie TalkNet: Is Someone Speaking? Exploring Long-term Temporal Features for Audio-visual Active Speaker Detection

XinzeLee This repository is based on Ultralytics/yolov5, with adjustments to enable polygon prediction boxes.

facebookresearch Per-Pixel Classification is Not All You Need for Semantic Segmentation

google Hardware accelerated, batchable and differentiable optimizers in JAX.

YugeTen This is the official PyTorch implementation of Gradient Matching for Domain Generalisation.

The-AI-Summer A collection of Deep Learning images and visualizations. The project has been developed by the AI Summer team and currently contains almost 100 images.

fabro66 Online 3D skeleton-based action recognition with a single RGB camera

ifsheldon PyTorch wrapper for Taichi data-oriented class

mindslab-ai WaveGrad 2: Unofficial PyTorch Implementation Iterative Refinement for Text-to-Speech Synthesis