Skip to content

This repository contains implementation of UNet, SegNet and DeepLab-v3+ using keras.

Notifications You must be signed in to change notification settings

shoaibsattar823/CS893-Semantic-Segmentation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

20 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Table of Content

Introduction

1. U-Net

2. SegNet

3. DeepLab-v3

CS893-Semantic-Segmentation

Semantic Segmentation is the classification of images on pixel level. The purpose is to understand and recognize what is in the image in pixel level. Semantic Segmentation has wide range of applications such as Robot Vision and Understanding, Autonomous Driving and Medical Purposes. In this project, I am doing semantic segmentation on MonuSeg Dataset using different segmentation techniques/architectures which are U-Net, Segnet and DeepLab-v3.

1. U-Net

(Modified the original implementation at https://github.com/hlamba28/UNET-TGS/blob/master/TGS%20UNET.ipynb for this dataset)

Model Diagram

model plot

Training Settings
  • Input Size: 1024x1024
  • Optimizer: SGD
  • Learning Rate: 0.001
  • Batch Size: 5
  • Epochs: 50
Evaluation
  • Accuracy = 0.82
  • Dice Score = 0.41
Learning Curve

Learning Curve

Visualization

demo1 demo2

2. SegNet

Model Diagram

model plot

Training Settings
  • Input: 256x256 size images for training which are actually patches of original 1000x1000 size image.
  • Optimizer: Adam
  • Batch Size: 16
  • Epochs: 50
Evaluation
  • Accuracy = 0.58
  • Dice Score = 0.35
  • F1-Score = 0.43
  • Loss (Binary Crossentropy) = 0.82
Learning Curve

Learning Curve Loss Learning Curve Acc

Visualization

demo1 demo2

3. DeepLab-v3

Model Diagram

model plot

Training Settings
  • Input Size: 256x256 patches from original 1000x1000 image
  • Optimizer: Adam
  • Batch Size: 8
  • Epochs: 32
Evaluation
  • Accuracy = 0.8968995
  • Dice Score = 0.74143386
  • F1-Score = 0.75504327
  • Loss (Binary Crossentropy) = 0.3610503834391397
Learning Curve

Learning Curve Loss

Visualization

demo1 demo2 demo3

About

This repository contains implementation of UNet, SegNet and DeepLab-v3+ using keras.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published