首先,您需要安装一些必要的库,如TensorFlow或Keras。然后,您可以使用以下代码来构建一个U-Net网络:
``` import tensorflow as tf from tensorflow.keras.layers import Input, Conv2D, MaxPooling2D, concatenate, UpSampling2D
inputs = Input((512, 512, 3))
down1 = Conv2D(64, (3, 3), padding='same')(inputs) down1 = Conv2D(64, (3, 3), padding='same')(down1) down1_pool = MaxPooling2D((2, 2), strides=(2, 2))(down1)
down2 = Conv2D(128, (3, 3), padding='same')(down1_pool) down2 = Conv2D(128, (3, 3), padding='same')(down2) down2_pool = MaxPooling2D((2, 2), strides=(2, 2))(down2)
down3 = Conv2D(256, (3, 3), padding='same')(down2_pool) down3 = Conv2D(256, (3, 3), padding='same')(down3) down3_pool = MaxPooling2D((2, 2), strides=(2,