使用Tensorflow怎么合并通道并加载子模型

今天就跟大家聊聊有关使用Tensorflow怎么合并通道并加载子模型，可能很多人都不太了解，为了让大家更加了解，小编给大家总结了以下内容，希望大家根据这篇文章可以有所收获。

（1）将两个子模型的输出合并到一个通道，例如同时连接到一个全连接层如图

合并方法为 tf.concat()函数。此函数需要两个个参数 concat（0或1，[合并节点1，合并节点2] ）。0 或 1 代表节点合并的方式：0 代表合并后列相同，行增加；1 代表合并后行相同，列增加。

上图所示合并方法为： X_20 = tf.concat(1, [X_top, X_down]);

（2）加载预训练好模型的部分参数。例如训练完成了一个五层网络，现在需要训练好一个七层网络，使用已训练好的五层网络参数初始化七层网络的前五层。

首先，五层网络保存模型时的参数变量名要和七层网络的需要初始化的参数变量名保持一致。

然后，定义加载指定变量名的 tf.train.Saver()。

例如：

Saver在restore模型时就会只加载 W_fc1_I, W_fc2_I, b_fc1_I, b_fc2_I 这四个参数，并且初始化当前模型中变量名相同的变量。需要注意的是，其他参数也是需要初始化的，一种避免遗漏初始化变量的方法是首先使用 initialize_all_variables() 函数对所有参数进行初始化之后再调用 restore 函数初始化需要的部分参数。

看完上述内容，你们对使用Tensorflow怎么合并通道并加载子模型有进一步的了解吗？如果还想了解更多知识或者相关内容，请关注亿速云行业资讯频道，感谢大家的支持。