lsh1994 / keras-segmentation

Get started with Semantic Segmentation based on Keras, including FCN32/FCN8/SegNet/U-Net
166 stars 55 forks source link
fcn keras segmentation

keras-segmentation

Implementation is not original papers. The purpose of this project is to get started with semantic segmentation and master the basic process.

FCN32/8、SegNet、U-Net Model published. Thank you for your support.

中文说明

Environment

Item Value Item Value
keras 2.2.4 OS win10
tensorflow-gpu 1.10/1.12 Python 3.6.7

Reference

https://github.com/divamgupta/image-segmentation-keras
https://github.com/ykamikawa/SegNet

Data:
https://drive.google.com/file/d/0B0d9ZiqAgFkiOHR1NTJhWVJMNEU/view?usp=sharing
Data or:
https://github.com/alexgkendall/SegNet-Tutorial/tree/master/CamVid

Project Strcutre

在这里插入图片描述

python visualizeDataset.py: Visual samples
在这里插入图片描述

python train.py: Execution train
python predict.py: Execution predict

You can modify the parameter in project switching model or cloning the historical version.

About

FCN32

Visualization results:
在这里插入图片描述

FCN8

Visualization results:
在这里插入图片描述

SegNet

U-Net