gpt4 book ai didi

c++ - 从 keras 保存 .pb 和 .pbtxt 文件

转载 作者:行者123 更新时间:2023-12-02 10:31:05 31 4
gpt4 key购买 nike

我需要得到 .pb.pbtxt来自 Keras model 的文件为了在 C++ 中使用经过训练的 Keras 模型和 OpenCV 的 DNN 模块。

我可以得到 .pb通过执行 model.save('model',save_format='tf') 文件但我还没有找到获得 .pbtxt 的方法文件。

最佳答案

我正在使用相同的设置并遇到相同的问题,所以这是我发现的:

  • 在 OpenCV 中,如果您使用 dnn::readNetFromTensorflow() 那么您实际上并不需要“.pbtxt”文件,仅“.pb”就足够了
  • 您使用 keras model.save() 方法生成的是一个 tensorflow SavedModel,它生成多个文件(以及 .pb),但 .pb 不能单独工作,OpenCV 不接受 SavedModel 格式。所以你需要生成一个卡住图。更多信息 here ,以及为 your model 生成卡住图的解决方案.
  • 关于c++ - 从 keras 保存 .pb 和 .pbtxt 文件,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/62288678/

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