Skip to content

Latest commit

 

History

History
178 lines (135 loc) · 9.16 KB

README.md

File metadata and controls

178 lines (135 loc) · 9.16 KB

Deep Learning For Dummies

Models

Deep Neural Networks

DNN

Model 1 (DNN)

  • Deep Neural Network created using TensorFlow/Keras and the MNIST dataset. 60,000 images each 28 pixels x 28 pixels with 10 possible classifications.
    • Layers
      • Input Layer: 784 Nodes
      • Hidden Layer 1: 512 Nodes
      • Hidden Layer 2: 512 Nodes
      • Output Layer: 10 Nodes
    • Details
      • Optimizer: adam
      • Loss Function: categorical_crossentropy
    • Accuracy
      • 98.12 %

Convolutional Neural Networks

CNN

Model 1 (CNN)

  • Convolutional Neural Network created using TensorFlow/Keras and the MNIST dataset. The weights are preloaded from a file due to the time taken for each epoch.
    • Layers
      • 28x28x1
        • Convolving: 5x5x32
      • 24x24x32
        • Pooling 2x2
      • 12x12x32
        • Convolving: 5x5x64
      • 8x8x64
        • Pooling: 2x2
      • 4x4x64 (Input)
      • Hidden Layer: 1024 Nodes
      • Output Layer: 10 Nodes
    • Details
      • Optimizer: adam
      • Loss Function: categorical_crossentropy
    • Accuracy
      • 99.30 %

VGG-16 Architecture

CNN2

Model 1 (VGG16)

  • VGG-16 Pretrained Predictor Model with ImageNet created using Keras with an imported version of VGG-16 and imported weights. It trains the model with the ImageNet dataset and enables you to pass an image into the model. It tries to predict what the image is and classify it based on the categories offered by ImageNet.

Model 2 (VGG16)

  • VGG16 Adaptations uses the CIFAR10 dataset. There are multiple models that are meant to mimic VGG16, but are slightly adapted due to the image sizes being 32x32 versus 244x244. I experiement with Convolutional Layers, Batch Normalization, Dropout, and Batch Size while trying to beat the imported VGG16 model.

Model 3 (VGG16)

  • VGG16 from Scratch uses PyTorch and the CIFAR10 Dataset. This is the best example of a Convolutional Neural Network implemented using PyTorch. Great reference for future models.

UNET Architecture

UNET

Model 1 (UNET)

  • Oxford Pets is a dataset with numerous types of pets and usesd a UNET architecture in TensorFlow to segment images.

Model 2 (UNET)

  • Carvana Image Segmentation is a dataset with numerous types of cars and is famous example for image segmentation. It uses PyTorch and an UNET architecture (no PyTorch Lightning).

Model 3 (UNET)

  • Oxford Pets Part 2 is using PyTorch Lightning! Make sure to add Jacard and Dice Number for better metrics.

Courses

Resources

TensorFlow

Keras

PyTorch

PyTorch Lightning