zhulf0804 / PSPNet-Tensorflow

A Tensorflow implementation of PSPNet.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

About

A PSPNet(Pyramid Scene Parsing Network) implementation with Tensorflow.

Set up

Exec

  • Train

    • for train + val dataset

      python train.py --dataset trainval

    • for train dataset

      python train.py

  • Inference [Use your trained model or download checkpoint here]

    • Inference an image in test set randomly

      python predict.py --prediction_on test

    • Inference an image in val set randomly

      python predict.py --prediction_on val

    • Inference an image in train set randomly

      python predict.py --prediction_on train

    • Inference an specified image by file path(or your own image path)

      python predict.py --file_path /Volumes/Samsung_T5/datasets/Cityscape/leftImg8bit_trainvaltest/leftImg8bit/test/berlin/berlin_000270_000019_leftImg8bit.png

  • Evaluation [Use your trained model or download checkpoint here]

    • On test set

      python evaluate.py --dataset test

    • On val set

      python evaluate.py --dataset val

Results

Desc Repo(%) Repo(%) Paper(%)
Train set train train+val train+val
mIoU 73.5 74.3 78.4

Prediction images

Pictures in test set.

ZhongGuancun Road in Beijing.

Tensorboard

  • cd summary directory and run the following command

    tensorboard --logdir=./

  • loss

  • learning rate

Reference

About

A Tensorflow implementation of PSPNet.


Languages

Language:Python 100.0%