unet-rgb

Keras(TF backend) implementation of unet for RGB pictures.

Overview

Requirement

• OpenCV
• Python 2.7
• Tensorflow-gpu
• Keras

Data

You can download all data here:https://pan.baidu.com/s/1WqmR-9jodyyUnLRCK6pcVw password: 533h

CamVid Data:https://github.com/preddy5/segnet/tree/master/CamVid

Howtouse

• extract downloaded data to corresponding directories
• run python data.py to generate 3 .npy files or you can download them to npydata
• run python unet.py to train, you can change hyperparameters on your own situation
• run python test2mask2pic.py to get test results(pictures)

Results

After training about 30 epochs, loss goes to about 0.005.The results seems OK! But the edges look a bit rough. I think that is Unet's own limitation.

Camvid

• run python data_camvid.py
• run python unet_camvid.py
• run python predict_camvid.py

After 50 epochs training, loss goes to about 4e-04 and acc is 0.9644.

About

Unet is mostly used in medical areas. I used this model for semantic segmentation of satellite remote sensing images in real work and the result is not bad. I think one of the reason is that it's a coarse-grained task like medical image analysis. Of course, enough labeled images are necessary.

Reference

https://github.com/zhixuhao/unet