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

张量流张量转换器模型是保存的模型还是frozengrpah?

如何解决张量流张量转换器模型是保存的模型还是frozengrpah?

我已经使用trt.TrtGraphConverterV2优化了对象检测模型。

converter = trt.TrtGraphConverterV2(
        input_saved_model_dir=input_saved_model_dir,conversion_params=conversion_params
    )
converter.convert()
 

,后来使用保存方法保存。它创建了一个目录,其中包含一个save_model.pb文件

converter.save(output_saved_model_dir=output_saved_model_dir)

当我尝试使用加载输出模型时

model_loaded = tf.saved_model.load_v2(input_saved_model_dir,tags=[tag_constants.SERVING]) 

正在加载,没有任何问题,但是我没有得到任何推断输出

我该如何加载模型以获取值。

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