cnv1989/deeplearning

This repo contains various deep learning model implementations

Lambdas/

Contains code that host trained models on AWS lambda. The lambdas are exposed via API gateway to enable classification trough APIs

Lambdas/Lenet5Lambda

Lenet5Lambda is implemented using Serverless Framework.

Lenet5/

Notebook

Tensorflow implementation of Lenet5 model for handwritten digit recognition.

Resnet/

Notebook

Training Script

Implented modified Resnet50 (by pruning the last residual block) for classifying CIFAR-10 dataset since resnet50 is built for 64x64 images while cifar-10 as has 32x32 images.

YOLO/

Notebook - Convert YAD2K model to Keras

Notebook - Evaluate YOLO

Takes YAD2K configuration and weights and converts to Keras. Applies box filtering and non max supression on the output of the model.