微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

如何从 tensorflow 2.x 中的冻结图efficentdet-dX.pb文件运行推理?

如何解决如何从 tensorflow 2.x 中的冻结图efficentdet-dX.pb文件运行推理?

我使用 automl pipeline 训练了一个 body 模型,该模型生成efficentdet-d4 文件,我已使用命令将其转换为 .ckpt(冻结图)文件.pb 存储库中给出。

automl

现在,我想在给定的图像上运行这个 python ./tensorflow/automl/efficientdet/model_inspect.py --runmode=saved_model --model_name=efficientdet-d4 --ckpt_path= ./models/model_d4_1/ --saved_model_dir=./frozen_graph/model_d4_1_80/ --hparams=./models/model_d4_1/config.yaml 文件。为此,我遵循 tensorflow object-detection-api docsofficial Guides 来运行模型并进行推理,但它抛出了 frozen-graph (efficentdet-d4.pb) 错误
请注意:如果我使用 TF2 Model Zoo 中给出的预训练网络,同样的代码运行良好。

'AutoTrackable' object is not callable

我尝试调试 automl 如何创建冻结图,以下是一些可能有用的关键路径: export_saved_model > build & export

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。