Unet++ 网络tensorflow版（slim实现）

最新推荐文章于 2024-09-07 20:30:50 发布

fegggye

最新推荐文章于 2024-09-07 20:30:50 发布

阅读量2.4k

点赞数 2

分类专栏：深度学习

本文链接：https://blog.csdn.net/u012160945/article/details/102852773

版权

本文详细介绍了如何使用TensorFlow的slim库实现Unet++网络结构，适用于图像分割任务。输入数据需归一化至[0,1]范围，并应用sigmoid激活得到0-1之间的预测值。附结构图展示网络工作原理。" 122135137,11753558,RocketMQ消费队列与索引文件恢复机制深度解析,"['Java', '后端开发', '消息队列']

摘要由CSDN通过智能技术生成

import tensorflow as tf
import numpy as np
import tensorflow.contrib.slim as slim
import os
from tensorflow.data import Dataset

reg = slim.l2_regularizer(scale=0.001) 

def standard_unit(input_tensor, stage, nb_filter, kernel_size=3):
    x = slim.conv2d(input_tensor, nb_filter, [3, 3], rate=1,activation_fn=None, weights_regularizer=reg)
    x = slim.batch_norm(x)
    x = tf.nn.relu(x)
    #x = slim.dropout(x)
    x = slim.conv2d(x, nb_filter, [3, 3], rate=1, scope=stage,activation_fn=None,weights_regularizer=reg)
    x = slim.batch_norm(x)
    x = tf.nn.relu(x)
    #x = slim.dropout(x)
    return x

def upsample(x,num_outputs,batch_size=1):
    pool_size = 2
    input_wh=int(x.shape[1])
    in_channels=int(x.shape[-1])
    output_shape=(batch_size,input_wh*2,input_wh*2,num_outputs)
    deconv_filter = tf.Variable(tf.truncated_normal([pool_size, pool_size, num_outputs, in_channels], stddev=0.02))
    deconv = tf.nn.conv2d_transpose(x, deconv_filter, output_shape, strides=[1, pool_size, pool_size, 1])
    return deconv


def UNet_pp(inputs, reg,deep_supervision=False):  # Unet
    '''

     1-1---> 1-2 ---> 1-3 ---> 1-4 --->1-5
        \   /   \   /    \    /   \   /
         2-1 --->2-2 ---> 2-3 --->2-4
           \    /   \    /   \   /
            3-1 --->