tf-ckpt-2-pb:将您的ckpt转换为pb savedModel以转换为tensorflow.js

时间:2024-05-29 12:33:35
【文件属性】:

文件名称:tf-ckpt-2-pb:将您的ckpt转换为pb savedModel以转换为tensorflow.js

文件大小:2KB

文件格式:ZIP

更新时间:2024-05-29 12:33:35

tensorflow Python

张量流ckpt到pb 编写了一个简单的cli应用程序,该应用程序将保存的tensorflow检查点转换为pb savedModel。 这背后的原因有时是我正在编写tensorflow.js程序,并且检查点保存的某些模型需要转换为 ,这没什么花哨的,只是我自己使用的简单工具 用法: python convert.py --checkpoint XX --model XX --out-path ./export -检查点 检查点的文件路径 - 模型 导出的 - 出去 输出目录,必须不存在 输出的保存模型将带有标签tfckpt2pb


【文件预览】:
tf-ckpt-2-pb-master
----.vscode()
--------settings.json(53B)
----convert.py(1KB)
----readme.md(720B)

网友评论