unet++的python代码

以下是一个简单的 UNet 模型的 Python 代码示例：

import tensorflow as tf
from tensorflow.keras.layers import Input, Conv2D, MaxPooling2D, Dropout, UpSampling2D, concatenate

def conv_block(inputs, filters, kernel_size, padding='same', activation='relu'):
    conv = Conv2D(filters=filters, kernel_size=kernel_size, padding=padding, activation=activation)(inputs)
    conv = Conv2D(filters=filters, kernel_size=kernel_size, padding=padding, activation=activation)(conv)
    return conv

def unet(input_size, num_classes=1):
    inputs = Input(input_size)

    # 编码器
    conv1 = conv_block(inputs, 64, 3)
    pool1 = MaxPooling2D(pool_size=(2, 2))(conv1)
    conv2 = conv_block(pool1, 128, 3)
    pool2 = MaxPooling2D(pool_size=(2, 2))(conv2)
    conv3 = conv_block(pool2, 256, 3)
    pool3 = MaxPooling2D(pool_size=(2, 2))(conv3)
    conv4 = conv_block(pool3, 512, 3)
    drop4 = Dropout(0.5)(conv4)

    # 解码器
    up5 = UpSampling2D(size=(2, 2))(drop4)
    up5 = concatenate([up5, conv3])
    conv5 = conv_block(up5, 256, 3)
    up6 = UpSampling2D(size=(2, 2))(conv5)
    up6 = concatenate([up6, conv2])
    conv6 = conv_block(up6, 128, 3)
    up7 = UpSampling2D(size=(2, 2))(conv6)
    up7 = concatenate([up7, conv1])
    conv7 = conv_block(up7, 64, 3)

    # 输出层
    outputs = Conv2D(num_classes, (1, 1), activation='sigmoid')(conv7)

    model = tf.keras.Model(inputs=inputs, outputs=outputs)

    return model

在这个示例代码中，我们定义了一个名为 `unet` 的函数，它接收输入大小和类别数作为参数，并返回一个 UNet 模型。模型包括一个编码器和一个解码器，其中编码器将输入图像压缩成一个较小的特征图，而解码器将这个特征图还原为图像。

在编码器中，我们使用了 4 个卷积块，每个块包括两个卷积层和一个最大池化层。在解码器中，我们使用了 3 个上采样层和 3 个卷积块，每个块也包括两个卷积层。最后，我们使用一个 $1\times1$ 的卷积层作为输出层，并使用 sigmoid 函数将每个像素映射到 $[0, 1]$ 的范围内。