梦沁清风

一、ckpt模型转换为frozen_graph.pb模型import tensorflow as tfdef freeze_graph(input_checkpoint, output_graph): output_node_names = "output" #获取的节点 saver = tf.train.import_meta_graph(input_checkpoin...

import tensorflow as tfimport numpy as npimport matplotlib.pyplot as pltfrom tensorflow.examples.tutorials.mnist import input_dataminst=input_data.read_data_sets("./mnist/",one_hot=True)one_mini...

import tensorflow as tfimport osfrom tensorflow.python import pywrap_tensorflowmodel_path="./checkPointModel/" # newModel checkPointModelinput_checkpoint=model_path+"model.ckpt"output_node_na...

 加入tf.contrib.quantize.create_train_graph()和tf.contrib.quantize.create_eval_graph()两个伪量化函数在训练代码中，属于完全量化的一个步骤，就是quantization aware training 方法，加入两句话是实现伪量化训练过程，自动添加伪量化节点，找到最大最小值，这样生成的pb文件是含有最大最小值的，模拟了...

一、mxnet转tensorflow模型1.下载mmdnn源码或pip安装，切换到mmdnn主目录下；安装时需要安装numpy+mkl的最新版本，准备好 mxnet模型的 .json文件和.params文件。2.运行命令行：python -m mmdnn.conversion._script.convertToIR -f mxnet -n model-symbol.json -w m...