Keras 使用 Lambda层详解
作者:驿无边 时间:2021-08-11 20:12:40
我就废话不多说了,大家还是直接看代码吧!
from tensorflow.python.keras.models import Sequential, Model
from tensorflow.python.keras.layers import Dense, Flatten, Conv2D, MaxPool2D, Dropout, Conv2DTranspose, Lambda, Input, Reshape, Add, Multiply
from tensorflow.python.keras.optimizers import Adam
def deconv(x):
height = x.get_shape()[1].value
width = x.get_shape()[2].value
new_height = height*2
new_width = width*2
x_resized = tf.image.resize_images(x, [new_height, new_width], tf.image.ResizeMethod.NEAREST_NEIGHBOR)
return x_resized
def Generator(scope='generator'):
imgs_noise = Input(shape=inputs_shape)
x = Conv2D(filters=32, kernel_size=(9,9), strides=(1,1), padding='same', activation='relu')(imgs_noise)
x = Conv2D(filters=64, kernel_size=(3,3), strides=(2,2), padding='same', activation='relu')(x)
x = Conv2D(filters=128, kernel_size=(3,3), strides=(2,2), padding='same', activation='relu')(x)
x1 = Conv2D(filters=128, kernel_size=(3,3), strides=(1,1), padding='same', activation='relu')(x)
x1 = Conv2D(filters=128, kernel_size=(3,3), strides=(1,1), padding='same', activation='relu')(x1)
x2 = Add()([x1, x])
x3 = Conv2D(filters=128, kernel_size=(3,3), strides=(1,1), padding='same', activation='relu')(x2)
x3 = Conv2D(filters=128, kernel_size=(3,3), strides=(1,1), padding='same', activation='relu')(x3)
x4 = Add()([x3, x2])
x5 = Conv2D(filters=128, kernel_size=(3,3), strides=(1,1), padding='same', activation='relu')(x4)
x5 = Conv2D(filters=128, kernel_size=(3,3), strides=(1,1), padding='same', activation='relu')(x5)
x6 = Add()([x5, x4])
x = MaxPool2D(pool_size=(2,2))(x6)
x = Lambda(deconv)(x)
x = Conv2D(filters=64, kernel_size=(3, 3), strides=(1,1), padding='same',activation='relu')(x)
x = Lambda(deconv)(x)
x = Conv2D(filters=32, kernel_size=(3, 3), strides=(1,1), padding='same',activation='relu')(x)
x = Lambda(deconv)(x)
x = Conv2D(filters=3, kernel_size=(3, 3), strides=(1, 1), padding='same',activation='tanh')(x)
x = Lambda(lambda x: x+1)(x)
y = Lambda(lambda x: x*127.5)(x)
model = Model(inputs=imgs_noise, outputs=y)
model.summary()
return model
my_generator = Generator()
my_generator.compile(loss='binary_crossentropy', optimizer=Adam(0.7, decay=1e-3), metrics=['accuracy'])
补充知识:含有Lambda自定义层keras模型,保存遇到的问题及解决方案
一,许多应用,keras含有的层已经不能满足要求,需要透过Lambda自定义层来实现一些layer,这个情况下,只能保存模型的权重,无法使用model.save来保存模型。保存时会报
TypeError: can't pickle _thread.RLock objects
二,解决方案,为了便于后续的部署,可以转成tensorflow的PB进行部署。
from keras.models import load_model
import tensorflow as tf
import os, sys
from keras import backend as K
from tensorflow.python.framework import graph_util, graph_io
def h5_to_pb(h5_weight_path, output_dir, out_prefix="output_", log_tensorboard=True):
if not os.path.exists(output_dir):
os.mkdir(output_dir)
h5_model = build_model()
h5_model.load_weights(h5_weight_path)
out_nodes = []
for i in range(len(h5_model.outputs)):
out_nodes.append(out_prefix + str(i + 1))
tf.identity(h5_model.output[i], out_prefix + str(i + 1))
model_name = os.path.splitext(os.path.split(h5_weight_path)[-1])[0] + '.pb'
sess = K.get_session()
init_graph = sess.graph.as_graph_def()
main_graph = graph_util.convert_variables_to_constants(sess, init_graph, out_nodes)
graph_io.write_graph(main_graph, output_dir, name=model_name, as_text=False)
if log_tensorboard:
from tensorflow.python.tools import import_pb_to_tensorboard
import_pb_to_tensorboard.import_to_tensorboard(os.path.join(output_dir, model_name), output_dir)
def build_model():
inputs = Input(shape=(784,), name='input_img')
x = Dense(64, activation='relu')(inputs)
x = Dense(64, activation='relu')(x)
y = Dense(10, activation='softmax')(x)
h5_model = Model(inputs=inputs, outputs=y)
return h5_model
if __name__ == '__main__':
if len(sys.argv) == 3:
# usage: python3 h5_to_pb.py h5_weight_path output_dir
h5_to_pb(h5_weight_path=sys.argv[1], output_dir=sys.argv[2])
来源:https://blog.csdn.net/a4775019136/article/details/99336591
标签:Keras,Lambda
0
投稿
猜你喜欢
Python学习之sys模块使用教程详解
2021-03-31 01:38:50
Python cookbook(数据结构与算法)找出序列中出现次数最多的元素算法示例
2021-04-16 10:50:05
对Python3 pyc 文件的使用详解
2023-05-02 01:54:57
python 对象和json互相转换方法
2022-06-27 18:40:14
用Python爬取LOL所有的英雄信息以及英雄皮肤的示例代码
2021-07-29 00:54:25
分析Python感知线程状态的解决方案之Event与信号量
2022-05-11 07:23:35
django框架模型层功能、组成与用法分析
2021-10-21 22:00:28
python单例模式之selenium driver实现单例
2021-09-30 14:31:03
Firefox Bug: inline/inline-block的间隙
2009-11-03 13:20:00
Python3 批量扫描端口的例子
2021-12-20 05:17:37
在python中将list分段并保存为array类型的方法
2023-11-15 10:18:00
Python容器类型公共方法总结
2022-06-24 23:58:06
基于Django的ModelForm组件(详解)
2022-03-07 18:21:04
详解PyTorch预定义数据集类datasets.ImageFolder使用方法
2022-01-30 20:15:07
MS IIS server Frontpage Ext Server漏洞
2008-05-04 09:54:00
Python数据可视化图实现过程详解
2022-08-30 19:24:17
PyTorch搭建CNN实现风速预测
2022-09-11 17:40:19
zen coding的dreamweaver插件安装教程
2010-01-12 13:30:00
一则python3的简单爬虫代码
2022-11-29 04:46:31
Python的函数的一些高阶特性
2022-06-04 15:06:28