1、动态获取
(1)朴素获取法
1) 朴素获取可训练变量:t_vars = tf.trainable_variables()
2)朴素获取全部变量,包含声明training=False变量:all_vars = tf.global_variables()
(2)使用tensorflow.contrib.slim
1) 获取常规变量(是slim里面与model变量对应的一个类型):regular_variables = slim.get_variables()
2)直接获取:vars = slim.get_variables_to_restore()
3)slim用于筛选方法
a. 通过name筛选: variables = slim.get_variables_by_name("d_")
b. 通过name后缀筛选:variables = slim.get_variables_by_suffix("_b")
c. 通过namespace筛选:variables = slim.get_variables(scope="layer1")
d. 通过include和exclude筛选
d0. variables_to_restore = slim.get_variables_to_restore(include=["d

本文介绍了在TensorFlow中动态获取模型变量列表的多种方式,包括使用`tf.trainable_variables`和`tf.global_variables`获取全部或训练变量,利用`tensorflow.contrib.slim`模块的`get_variables`和`get_variables_to_restore`进行筛选,以及如何从已保存的模型文件中离线获取变量列表。

4119

被折叠的 条评论
为什么被折叠?



