This code uses Fully Convolutional Networks in Tensorflow to solve the semantic segmentation problem. The post traininf process is not included.
Deconvolution Layers are initialized as bilinear upsampling. Conv and FCN layer weights using VGG weights. Numpy load is used to read VGG weights.The .npy file for VGG16 however need to be downloaded before using this needwork.
use readimage.py for image storage.
run train_pipeline.py for training process.
change the checkpoint path in test.py and run it for testing
(there're already 200 pic in the test_data folder for reference)
tensorflow 0.10
keras
skimage
cv2
numpy
matplotlib