由于项目需要,需要将TensorFlow保存的模型从ckpt文件转换为pb文件。
import os
from tensorflow.python import pywrap_tensorflow
from net2use import inception_resnet_v2_small#这里使用自己定义的模型函数即可
import tensorflow as tf
if __name__=='__main__':
pb_file = "./model/output.pb"
ckpt_
在使用TensorFlow的过程中,保存模型参数变量是很重要的一个环节,既可以保证训练过程信息不丢失,也可以帮助我们在需要快速恢复或使用一个模型的时候,利用之前保存好的参数之间导入,可以节省大量的训练时间。本文通过最简单的例程教大家如何保存和读取.ckpt文件。
一、保存到文件
首先是导入必要的东西:
import tensorflow as tf
import numpy as np
随便写几个变量:
# Save to file
# remember to define the sam