unet的tensorflow代码_基于Tensorflow的Unet网络实现

论文地址：http://www.arxiv.org/pdf/1505.04597.pdf

Unet网络，分割领域的经典之作，大家可以尝试一下。废话少说，上代码。

import tensorflow as tf

def convolutional(input_data, filters_shape, trainable, name, downsample=False, activate=True, bn=True):

with tf.variable_scope(name):

if downsample:

pad_h, pad_w = (filters_shape[0] - 2) // 2 + 1, (filters_shape[1] - 2) // 2 + 1

paddings = tf.constant([[0, 0], [pad_h, pad_h], [pad_w, pad_w], [0, 0]])

input_data = tf.pad(input_data, paddings, 'CONSTANT')

strides = (1, 2, 2, 1)

padding = 'VALID'

else:

strides = (1, 1, 1, 1)

padding = "SAME"

weight = tf.get_variable(name='weight', dtype=tf.float32, trainable=True,

shape=filters_shape, initializer&