tensorflow pretrained model
来源:互联网 发布:.net framework mac版 编辑:程序博客网 时间:2024/05/16 02:52
tensorflow 简单的加载训练好的模型,如果某一层形状不一样就自动的省略,也可以用reshape。
def optimistic_restore(session, save_file): reader = tf.train.NewCheckpointReader(save_file) saved_shapes = reader.get_variable_to_shape_map() var_names = sorted([(var.name, var.name.split(':')[0]) for var in tf.global_variables() if var.name.split(':')[0] in saved_shapes]) restore_vars = [] with tf.variable_scope('', reuse=True): for var_name, saved_var_name in var_names: curr_var = tf.get_variable(saved_var_name) var_shape = curr_var.get_shape().as_list() if var_shape == saved_shapes[saved_var_name]: restore_vars.append(curr_var) saver = tf.train.Saver(restore_vars) saver.restore(session, save_file)
reference
https://github.com/tensorflow/tensorflow/issues/312
阅读全文
1 0
- tensorflow pretrained model
- deep learning pretrained model
- Caffe shape mismatch error using pretrained VGG-16 model
- 使用mxnet的预训练模型(pretrained model)分类与特征提取
- Tensorflow Model Persistence
- Tensorflow-save model
- Tensorflow-model模板
- Tensorflow-model API
- Build basic model in Tensorflow
- 为什么“Pretrained+Fine-tuning”
- tensorflow学习笔记(十一):seq2seq Model
- ResNet-TensorFlow Model Zoo代码理解
- TensorFlow学习笔记8----TensorFlow Linear Model Tutorial
- RNN代码解读之char-RNN with TensorFlow(model.py)
- CS 20SI|Lecture 4 Structure your TensorFlow model
- Tensorflow: How to restore a previously saved model (python)
- Tensorflow使用pip安装后没有model目录
- 利用opencv3读取tensorflow model,对图像进行分类
- 怎么解决VMware“该虚拟机似乎正在使用中”问题
- cv_bridge与OpenCV版本问题
- Birt报表上传到Linux服务器上,打印PDF乱码问题
- web sql使用
- 阿里云CentOS7挂载SSD云盘的方法
- tensorflow pretrained model
- 代码计算抛物线
- Linux_04 权限管理
- Androidpn里的Xmpp的理解
- 记录百度地图接入的方法
- PI表中新增字段
- Hadoop实践(二)---集群和开发环境搭建(Intellij IDEA & Maven 开发Hadoop)
- uImage生成过程
- 洛谷P1003 铺地毯(简单模拟)