20241022_01

from keras import Input

from keras.layers import Conv2D

from keras.layers import MaxPooling2D

from keras.layers import Dropout

from keras.models import Model

from keras.layers import concatenate

from tensorflow.keras.optimizers import Adam

from keras.layers import Conv2DTranspose

def Enhancednet(pretrained_weights=None):

input_shape = (None, None, 1)

inputs = Input(shape=input_shape, name='input_img')

conv1 = Conv2D(16, 5, activation='relu', padding='same')(inputs)

drop1 = Dropout(0.6)(conv1)

pool1 = MaxPooling2D(pool_size=(2, 2))(drop1)

conv2 = Conv2D(24, 5, activation='relu', padding='same')(pool1)

drop2 = Dropout(0.6)(conv2)

pool2 = MaxPooling2D(pool_size=(2, 2))(drop2)

conv3 = Conv2D(32, 5, activation='relu', padding='same')(pool2)

drop3 = Dropout(0.6)(conv3)

pool3 = MaxPooling2D(pool_size=(2, 2))(drop3)

conv4 = Conv2D(40, 5, activation='relu', padding='same')(pool3)

drop4 = Dropout(0.6)(conv4)

up5 = Conv2D(32, 3, activation='relu', padding='same')(

Conv2DTranspose(32, 5, activation='relu', padding="same", strides=2)(drop4))

merge5 = concatenate([drop3, up5], axis=3)

conv5 = Conv2D(32, 5, activation='relu', padding='same')(merge5)

drop5 = Dropout(0.6)(conv5)

up6 = Conv2D(24, 3, activation='relu', padding='same')(

Conv2DTranspose(24, 5, activation='relu', padding="same", strides=2)(drop5))

merge6 = concatenate([drop2, up6], axis=3)

conv6 = Conv2D(24, 5, activation='relu', padding='same')(merge6)

drop6 = Dropout(0.6)(conv6)

up7 = Conv2D(16, 3, activation='relu', padding='same')(

Conv2DTranspose(16, 5, activation='relu', padding="same", strides=2)(drop6))

merge7 = concatenate([drop1, up7], axis=3)

conv7 = Conv2D(16, 5, activation='relu', padding='same')(merge7)

drop7 = Dropout(0.6)(conv7)

conv8 = Conv2D(1, 1, activation='relu')(drop7)

model = Model(inputs=inputs, outputs=conv8)

opt = Adam()

model.compile(optimizer=opt, loss='mse', metrics=['accuracy'])

if pretrained_weights:

model.load_weights(pretrained_weights)

return model

相关推荐
江沉晚呤时4 小时前
在 C# 中调用 Python 脚本:实现跨语言功能集成
python·microsoft·c#·.net·.netcore·.net core
电脑能手5 小时前
如何远程访问在WSL运行的Jupyter Notebook
ide·python·jupyter
Edward-tan5 小时前
CCPD 车牌数据集提取标注,并转为标准 YOLO 格式
python
老胖闲聊6 小时前
Python I/O 库【输入输出】全面详解
开发语言·python
倔强青铜三6 小时前
苦练Python第18天:Python异常处理锦囊
人工智能·python·面试
倔强青铜三6 小时前
苦练Python第17天:你必须掌握的Python内置函数
人工智能·python·面试
迷路爸爸1806 小时前
让 VSCode 调试器像 PyCharm 一样显示 Tensor Shape、变量形状、变量长度、维度信息
ide·vscode·python·pycharm·debug·调试
咸鱼鲸7 小时前
【PyTorch】PyTorch中的数据预处理操作
人工智能·pytorch·python
Dxy12393102167 小时前
Python ExcelWriter详解:从基础到高级的完整指南
开发语言·python
金玉满堂@bj8 小时前
Conda 安装包的用途
python