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

时间:2020-10-29 12:13:48

标签: deep-learning object-detection inference tensorflow1.15 tensorrt-python

我已经使用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]) 

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

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

0 个答案:

没有答案
相关问题