Tensorflow的ckpt转为npy格式的代码

由于无法把非神经网络分类器在tensorflow框架中搭建，想把神经网络的输出概率用到其他的非神经网络的分类其中，这就需要把神经网络中保存的参数提取出来。由于神经网络是基于图的计算，有自己的保存方式，我们不能随意保存和提取其中的参数。下面是使用一个pywrap_tensorflow的工具去读ckpt文件，将其中保存的信息读取出来，再保存到我们想要的格式。

from tensorflow.python import pywrap_tensorflow
import numpy as np
checkpoint_path="/tmp/prop/softmax_out.ckpt"
reader=pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
var_to_shape_map=reader.get_variable_to_shape_map()
param =[]for key in var_to_shape_map:print ("tensor_name",key)param.append(reader.get_tensor(key))np.save('dnnout.npy',param)

Tensorflow的ckpt转为npy格式的代码相关推荐

通过输入流获取json格式数据转为map格式
将通过输入流获取json格式数据转为map格式转换代码: try {int length = 0;//通过输入流获取到json格式数据ServletInputStream inputStream = ...
python快速批量将jpg图片格式转为pgm格式
python快速批量将jpg或者png图片格式转为pgm格式. 代码里两个部分要改成你自己的路径,已经在代码备注. import os import cv2path = r'D:\DeepLearni ...
tensorflow的ckpt文件转为npy文件
我们迁移模型参数时,需要将ckpt文件转为npy文件,以VGG19为例: """ 将ckpt文件转化为npy文件 """import num ...
Tensorflow入门——自制数据集：将未经处理的图片制成npy格式的数据集
一些吐槽和文章简介博主是那种"拿来就用,不会再查"的人.而几乎所有教程都秉持着"这应该是python课教的"的理念,从而使得没学过python的人颇为头疼.博 ...
python浪漫代码表白npy_Python 存取npy格式数据实例
数据处理的时候主要通过两个函数 (1):np.save("test.npy",数据结构) ----存数据 (2):data =np.load('test.npy") -- ...
记录h5文件数据转为npy和csv格式类型的方法
最近做数据分析,需要把h5文件转成npy,然后再转成csv格式,因此记录一下方法. 首先利用Pandas读取.h5数据文件, import pandas as pd import numpy as n ...
TensorFlow训练参数存为npy格式并调用——线性回归
模型训练并保存 # -*- coding: utf-8 -*- """ Created on Sun Mar 15 10:27:32 2020@author: weixi ...
Python模块介绍使用：img2pdf 几行代码将图片转为PDF格式文件
hello,大家好,今天我来推荐一个模块--img2pdf,这个模块可以把图片转换为pdf,开始学习吧! 1. pip安装使用命令: pip install img2pdf 安装完毕后,打开一个Py ...
ckpt转npy、转csv、转pickle详解
因为程序里的参数需要直接加载训练好的权重值,在此基础上再进行训练,所以需要在另一个程序里训练好权重,先将其保存到本地,然后该程序再来加载使用. 我们知道,通过以下代码,在训练过程中会得到checkpo ...
python转csv_python脚本如何将Excel文件转为csv文件（代码）
本篇文章给大家带来的内容是关于python脚本如何将Excel文件转为csv文件(代码),有一定的参考价值,有需要的朋友可以参考一下,希望对你有所帮助.#!/usr/bin/env python __ ...

Tensorflow的ckpt转为npy格式的代码

Tensorflow的ckpt转为npy格式的代码相关推荐

最新文章

热门文章