Stars
Minimal implementation of clipped objective Proximal Policy Optimization (PPO) in PyTorch
📊 Save matplotlib figures as TikZ/PGFplots for smooth integration into LaTeX.
Entity Abstraction in Visual Model-Based Reinforcement Learning
Simple and easily configurable 3D FPS-game-like environments for reinforcement learning
Code repository for the Animal AI Olympics competition
A PyTorch implementation of MobileNet V2 architecture and pretrained model.
Learning Confidence for Out-of-Distribution Detection in Neural Networks
3D Unet Equipped with Advanced Deep Learning Methods
Pytorch 3D U-Net Convolution Neural Network (CNN) designed for medical image segmentation
Lung fields segmentation on tomography images using convolutional neural networks.
Image-to-image translation in PyTorch (e.g., horse2zebra, edges2cats, and more)
An Automatic Model Compression (AutoMC) framework for developing smaller and faster AI applications.
This repo is an implementation of quantized CNN for both weights (1-bit compression) and feature maps (2-bit compression).
Code for SkipNet: Learning Dynamic Routing in Convolutional Networks (ECCV 2018)
Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distiller
BlockDrop: Dynamic Inference Paths in Residual Networks
Cayman is a Jekyll theme for GitHub Pages
Progressive Growing of GANs for Improved Quality, Stability, and Variation
Collection of generative models, e.g. GAN, VAE in Pytorch and Tensorflow.