gpt4 book ai didi

tensorflow - 如何将 .ckpt 文件转换为 .pb

转载 作者:行者123 更新时间:2023-12-04 01:45:45 26 4
gpt4 key购买 nike

我在对象检测 API 中使用 ssd_mobilenets 来训练我自己的模型,并获取 .ckpt 文件。它在我的电脑上运行良好,但现在我想在我的手机上使用该模型。所以,我需要将它转换为 .pb 文件。我不知道该怎么做,有人可以帮忙吗?顺便说一下,ssd_mobilenets 的图表很复杂,我找不到哪个是模型的输出。有没有人知道输出的名称?

最佳答案

使用export_inference_graph.py将模型检查点文件转换为 .pb 文件。

python tensorflow_models/object_detection/export_inference_graph.py \
--input_type image_tensor \
--pipeline_config_path architecture_used_while_training.config \
--trained path_to_saved_ckpt/model.ckpt-NUMBER \
--output_directory model/

关于tensorflow - 如何将 .ckpt 文件转换为 .pb,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45726365/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com