现在位置: 首页 > CheckPoint
2020年02月18日 编程语言 ⁄ 共 1388字 评论关闭

TensorFlow实现checkpoint文件转换为pb文件 由于项目需要,需要将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"

阅读全文