pbsaveas()函数用于将当前图表保存为指定格式的文件。该函数有三个参数:
1. filename:要保存的文件名,可以是绝对路径或相对路径。
2. format:要保存的文件格式,可以是png,jpg,eps,pdf等。
3. dpi:图片的分辨率,默认为100。
示例:
pbsaveas('test.png','png',200)
通常训练模型的时候是保存ckpt方便接着训练,但是上线可以保存为pb模型,加载的时候不需要重新定义模型,只用输入输出来调用模型。
save_model 和load_model两个函数要分开执行,第一次注释掉load,只save,第二次load的时候注释掉save。因为声明模型的时候都是用默认图,变量命名会依次是0,1,load的时候名字对应不上。
保存好的pb模型路径文件格式为
还有一种保存pb模型的方法,保存的东西只有model.pb没有variables。
如果需要发布模型成服务,请看: https://www.jianshu.com/p/5b74f1bc0178
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)