当前位置：首页 > news >正文

tensorflow的unet模型

news 2025/7/24 5:31:57

import tensorflow as tf
from tensorflow.keras.layers import Input, Conv2D, MaxPooling2D, Dropout, UpSampling2D, concatenate# 定义 U-Net 模型
def unet(input_size=(256, 256, 3)):inputs = Input(input_size)# 编码器部分conv1 = Conv2D(64, 3, activation='relu', padding='same', kernel_initializer='he_normal')(inputs)conv1 = Conv2D(64, 3, activation='relu', padding='same', kernel_initializer='he_normal')(conv1)pool1 = MaxPooling2D(pool_size=(2, 2))(conv1)conv2 = Conv2D(128, 3, activation='relu', padding='same', kernel_initializer='he_normal')(pool1)conv2 = Conv2D(128, 3, activation='relu', padding='same', kernel_initializer='he_normal')(conv2)pool2 = MaxPooling2D(pool_size=(2, 2))(conv2)conv3 = Conv2D(256, 3, activation='relu', padding='same', kernel_initializer='he_normal')(pool2)conv3 = Conv2D(256, 3, activation='relu', padding='same', kernel_initializer='he_normal')(conv3)pool3 = MaxPooling2D(pool_size=(2, 2))(conv3)conv4 = Conv2D(512, 3, activation='relu', padding='same', kernel_initializer='he_normal')(pool3)conv4 = Conv2D(512, 3, activation='relu', padding='same', kernel_initializer='he_normal')(conv4)drop4 = Dropout(0.5)(conv4)pool4 = MaxPooling2D(pool_size=(2, 2))(drop4)# 中间层conv5 = Conv2D(1024, 3, activation='relu', padding='same', kernel_initializer='he_normal')(pool4)conv5 = Conv2D(1024, 3, activation='relu', padding='same', kernel_initializer='he_normal')(conv5)drop5 = Dropout(0.5)(conv5)# 解码器部分up6 = Conv2D(512, 2, activation='relu', padding='same', kernel_initializer='he_normal')(UpSampling2D(size=(2, 2))(drop5))merge6 = concatenate([drop4, up6], axis=3)conv6 = Conv2D(512, 3, activation='relu', padding='same', kernel_initializer='he_normal')(merge6)conv6 = Conv2D(512, 3, activation='relu', padding='same', kernel_initializer='he_normal')(conv6)up7 = Conv2D(256, 2, activation='relu', padding='same', kernel_initializer='he_normal')(UpSampling2D(size=(2, 2))(conv6))merge7 = concatenate([conv3, up7], axis=3)conv7 = Conv2D(256, 3, activation='relu', padding='same', kernel_initializer='he_normal')(merge7)conv7 = Conv2D(256, 3, activation='relu', padding='same', kernel_initializer='he_normal')(conv7)up8 = Conv2D(128, 2, activation='relu', padding='same', kernel_initializer='he_normal')(UpSampling2D(size=(2, 2))(conv7))merge8 = concatenate([conv2, up8], axis=3)conv8 = Conv2D(128, 3, activation='relu', padding='same', kernel_initializer='he_normal')(merge8)conv8 = Conv2D(128, 3, activation='relu', padding='same', kernel_initializer='he_normal')(conv8)up9 = Conv2D(64, 2, activation='relu', padding='same', kernel_initializer='he_normal')(UpSampling2D(size=(2, 2))(conv8))merge9 = concatenate([conv1, up9], axis=3)conv9 = Conv2D(64, 3, activation='relu', padding='same', kernel_initializer='he_normal')(merge9)conv9 = Conv2D(64, 3, activation='relu', padding='same', kernel_initializer='he_normal')(conv9)outputs = Conv2D(1, 1, activation='sigmoid')(conv9)model = tf.keras.Model(inputs=inputs, outputs=outputs)return model# 创建 U-Net 模型
model = unet()
model.summary()

一个简单的 U-Net 模型的 TensorFlow 2.x 代码示例。U-Net 模型通常用于图像分割任务。

import tensorflow as tf
import numpy as np
from tensorflow.keras.layers import Input, Conv2D, MaxPooling2D, Dropout, UpSampling2D, concatenate
from tensorflow.keras.models import Model
from tensorflow.keras.optimizers import Adam
from sklearn.model_selection import train_test_split# 定义 U-Net 模型
def unet(input_size=(256, 256, 3)):inputs = Input(input_size)# 编码器部分...# 解码器部分...# 输出层...#请看上述代码model = Model(inputs=inputs, outputs=outputs)return model# 准备训练数据和标签
# 这里假设你已经有了医学图像数据集，以及相应的分割标签
# 请根据你的数据集的格式来加载和预处理数据
# 假设训练数据和标签分别存储在 train_images 和 train_masks 中# 划分训练集和验证集
X_train, X_val, y_train, y_val = train_test_split(train_images, train_masks, test_size=0.2, random_state=42)# 创建 U-Net 模型
model = unet()# 编译模型
model.compile(optimizer=Adam(lr=1e-4), loss='binary_crossentropy', metrics=['accuracy'])# 训练模型
model.fit(X_train, y_train, batch_size=4, epochs=10, validation_data=(X_val, y_val))# 保存模型
model.save("unet_medical_segmentation.h5")

在加载和预处理医学图像数据时，可以使用 Python 中的图像处理库如 OpenCV 或 Pillow。调整大小和预处理医学图像的示例代码

import cv2
import numpy as np
def load_and_preprocess_image(image_path, target_size=(256, 256)):# 读取图像image = cv2.imread(image_path)# 如果需要，调整图像大小if target_size is not None:image = cv2.resize(image, target_size)# 预处理图像（根据你的需求进行进一步的预处理，如归一化）image = image / 255.0  # 归一化到 [0, 1] 范围return image
# 示例用法：
image_path = "your_image.jpg"  # 替换成你的图像文件路径
target_size = (256, 256)  # 替换成你想要的目标大小
# 加载和预处理图像
preprocessed_image = load_and_preprocess_image(image_path, target_size)
# 打印图像的形状（用于验证）
print("Image shape:", preprocessed_image.shape)

它遍历目录中的图像文件，加载并预处理这些图像，然后将它们转换为 TensorFlow 张量

import tensorflow as tf
import os
import cv2
import numpy as npdef load_and_preprocess_image(image_path, target_size=(256, 256)):# 读取图像image = cv2.imread(image_path)# 如果需要，调整图像大小if target_size is not None:image = cv2.resize(image, target_size)# 预处理图像（根据你的需求进行进一步的预处理，如归一化）image = image / 255.0  # 归一化到 [0, 1] 范围return imagedef load_and_preprocess_images_in_directory(directory, target_size=(256, 256)):images = []# 遍历目录中的图像文件for filename in os.listdir(directory):if filename.endswith(".jpg") or filename.endswith(".png"):image_path = os.path.join(directory, filename)preprocessed_image = load_and_preprocess_image(image_path, target_size)images.append(preprocessed_image)# 将图像列表转换为 TensorFlow 张量images = np.array(images)images = tf.convert_to_tensor(images, dtype=tf.float32)return images# 示例用法：
image_directory = "your_image_directory"  # 替换成包含图像的目录路径
target_size = (256, 256)  # 替换成你想要的目标大小# 加载和预处理图像
image_tensor = load_and_preprocess_images_in_directory(image_directory, target_size)# 打印图像张量的形状（用于验证）
print("Image tensor shape:", image_tensor.shape)

查看全文

http://www.lryc.cn/news/170030.html