如何解决张量流张量转换器模型是保存的模型还是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 举报,一经查实,本站将立刻删除。