ykamikawa / tf-keras-SegNet

SegNet including indices pooling for Semantic Segmentation with tensorflow and keras

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

SegNet

SegNet is a model of semantic segmentation based on Fully Comvolutional Network.

This repository contains the implementation of learning and testing in keras and tensorflow. Also included is a custom layer implementation of index pooling, a new property of segnet.

architecture

  • encoder decoder architecture

  • fully convolutional network

  • indices pooling

    indicespooling

description

This repository is SegNet architecture for Semantic Segmentation. The repository of other people's segmentation, pooling with indices not implemented.But In this repository we implemented pooling layer and unpooling layer with indices at MyLayers.py.

Segnet architecture is early Semantic Segmentation model,so acccuracy is low but fast. In the future, we plan to implement models with high accuracy.(UNet,PSPNet,Pix2Pix ect..)

Usage

train

python SegNet.py [--options your dataset]

About

SegNet including indices pooling for Semantic Segmentation with tensorflow and keras


Languages

Language:Python 100.0%