本实验基于SegNet网络进行实验，用于钢铁分割

SegNet

SegNet implementation in Tensorflow.

Environments

• Python 3.6.2
• Numpy==1.14.3
• Pillow==4.2.1
• tensorflow-gpu==1.3.0
• tensorflow-tensorboard==0.1.5

Usage

1. Dataset ''' Referecne the last github https://github.com/fourmi1995/IronData_CamvidStyle

2. Convert CamVid dataset to TFRecord format

python camvid.py --target train

3. Training

python train.py \
  --iteration 20000 \
  --snapshot 4000 \
  --optimizer adadelta \
  --learning_rate 1.0

4. Evaluation

python eval.py \
  --resdir eval \
  --outdir output/camvid/segnet \
  --checkpoint_dir output/camvid/segnet/trained_model \
  --num_sample 233

Reference

• https://github.com/Pepslee/tensorflow-contrib
• https://github.com/warmspringwinds/tf-image-segmentation
• https://github.com/fabianbormann/Tensorflow-DeconvNet-Segmentation
• https://github.com/eragonruan/refinenet-image-segmentation
• https://github.com/DrSleep/tensorflow-deeplab-resnet
• https://github.com/aizawan/segnet 原github:https://github.com/tkuanlun350/Tensorflow-SegNet