Machine Learning Paper Implementations

by Jeffrey Tan

Architectures

A Better Autoencoder for Image: Convolutional Autoencoder

tanjeffreyz/convolutional-autoencoder
autoencoder convolutional-neural-networks lfw pytorch

Pytorch implementation of "A Better Autoencoder for Image: Convolutional Autoencoder" by Yifei Zhang

Gradient-Based Learning Applied to Document Recognition (LeNet-5)

tanjeffreyz/lenet-5
pytorch mnist deep-learning convolutional-networks

PyTorch implementation of LeNet-5 published in "Gradient-Based Learning Applied to Document Recognition" by Y. Lecun, L. Bottou, Y. Bengio, P. Haffner

You Only Look Once: Unified, Real-Time Object Detection (YOLOv1)

tanjeffreyz/yolo-v1
object-detection pytorch yolo computer-vision non-maximum-suppression

PyTorch implementation of the YOLOv1 architecture presented in "You Only Look Once: Unified, Real-Time Object Detection" by Joseph Redmon, Santosh Divvala, Ross Girshick, Ali Farhadi

A Neural Algorithm of Artistic Style

tanjeffreyz/neural-algorithm-of-artistic-style

PyTorch implementation of "A Neural Algorithm of Artistic Style" by Leon A. Gatys, Alexander S. Ecker, and Matthias Bethge

Generative Adversarial Networks

tanjeffreyz/generative-adversarial-networks
deep-learning generative-adversarial-network pytorch cifar10 mnist

PyTorch implementation of "Generative Adversarial Networks" by Ian J. Goodfellow, Jean Pouget-Abadie, Mehdi Mirza, Bing Xu, David Warde-Farley, Sherjil Ozair, Aaron Courville, Yoshua Bengio

Deep Residual Learning for Image Recognition

tanjeffreyz/deep-residual-learning
pytorch cifar-10 deep-learning residual-networks convolutional-networks

PyTorch implementations of the deep residual networks published in "Deep Residual Learning for Image Recognition" by Kaiming He, Xiangyu Zhang, Shaoqing Ren, Jian Sun

Techniques

Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift

tanjeffreyz/batch-normalization
batch-normalization internal-covariate-shift pytorch mnist

PyTorch implementation of batch normalization from "Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift" by Sergey Ioffe, Christian Szegedy