我已经使用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])
正在加载,没有任何问题,但是我没有得到任何推断输出。
我该如何加载模型以获取值。