keras-PSPNet

Pyramid Scene persing Network is a model of semantic segmentation based on Fully Comvolutional Network. This repository contains the implementation of learning and testing in keras and tensorflow.

Architecutre

• atrous convalution

• residual module

• pyramid pooling module

Prerequirements

• python3.6

• opencv for python

• keras,tensorflow

Usage

train

• Segmentation involveing multiple categories

  python train.py --options

• Segmentation of mask image

  python train_mask.py --options

• image dir

• mask image dir

• batchsize, nb_epochs, epoch_per_steps, input_configs

• class weights

• device num

• options

test

• Input test image

• responce json format involving category name and color(pixel based prediction)

  python predict.py --input_path [path/to//input_imahge]