Skip to content

Vertical-Beach/segmentation_unet

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

49 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Semantic Segmentation using U-Net on Pascal VOC 2012

This repository implements semantic segmentation on Pascal VOC2012 using U-Net.

An article about this implementation is here.

Semantic segmentation is a kind of image processing as below.

This package includes modules of data loader, reporter(creates reports of experiments), data augmenter, u-net model, and training it.

Usage

To show how to run.

python main.py --help

To run with data augmentation using GPUs.

python main.py --gpu --augmentation

U-Net

U-Net is an encoder-decoder model consisted of only convolutions, without fully connected layers.

U-Net has a shape like "U" as below, that's why it is called U-Net.

Experiments

The following results is got by default settings.

Results of segmentation

For the training set

training1

training2

For the test set

test1

test2

test3

Accuracy and Loss

About

Semantic segmentation using U-NET

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%