martinengelcke

PRML

PRML algorithms implemented in Python

tex

This repository provides convenient and readable macros for commonly used syntax when writing scientific articles about machine learning.

dspn

[NeurIPS 2019] Deep Set Prediction Networks

yolact

A simple, fully convolutional model for real-time instance segmentation.

pytorch-segmentation-toolbox

PyTorch Implementations for DeeplabV3 and PSPNet

IODINE

PyTorch re-implementation of Multi-Object Representation Learning with Iterative Variational Inference

PyRep

A toolkit for robot learning research.

DALI

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

pytorch-segmentation

:art: Semantic segmentation models, datasets and losses implemented in PyTorch.

pytorch-deeplab-xception

DeepLab v3+ model in PyTorch. Support different backbones.

pytorch-semantic-segmentation

PyTorch for Semantic Segmentation

OP3

Entity Abstraction in Visual Model-Based Reinforcement Learning

linearized_multisampling_release

Code release for ICCV 2019 Oral, "Linearized Multi-Sampling for Differentiable Image Transformation"

lmdis-rep

"Unsupervised Discovery of Object Landmarks as Structural Representations" Project page: http://www.ytzhang.net/projects/lmdis-rep/

inverse-compositional-STN

Inverse Compositional Spatial Transformer Networks :performing_arts: (CVPR 2017 oral)

NAS-Bench-201

NAS-Bench-201 API and Instruction

SceneNetv1.0

Still a work in progress and adding code..

vision

Datasets, Transforms and Models specific to Computer Vision

rcnn-depth

Learning Rich Features from RGB-D Images for Object Detection and Segmentation

fcn.berkeleyvision.org

Fully Convolutional Networks for Semantic Segmentation by Jonathan Long*, Evan Shelhamer*, and Trevor Darrell. CVPR 2015 and PAMI 2016.

SB-GAN

Semantic Bottleneck Scene Generation

c-swm

Contrastive Learning of Structured World Models

Matterport

Matterport3D is a pretty awesome dataset for RGB-D machine learning tasks :)

pytorch-semseg

Semantic Segmentation Architectures Implemented in PyTorch

AlphaPose

Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System

Batch-Renormalization-PyTorch

Batch Renormalization module for PyTorch based on paper https://arxiv.org/abs/1702.03275 .

votenet

Deep Hough Voting for 3D Object Detection in Point Clouds

pixelsnail-public

VQ-VAE

Minimalist implementation of VQ-VAE in Pytorch

vq-vae-2-pytorch

Implementation of Generating Diverse High-Fidelity Images with VQ-VAE-2 in PyTorch