上传者: 38725531
|
上传时间: 2022-11-02 15:59:21
|
文件大小: 37KB
|
文件类型: PDF
由于项目需要,需要将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_file = "./model/model.ckpt-652900"
'''
这里的节点名字可能