Tensorrt筆記(四)推理分割模型

juluwangriyue 2022-06-23 11:02:33 阅读数:895

tensorrt推理分割模型

Tensorrt筆記(四)推理分割模型

生成trt模型後,使用python脚本測試推理結果。

def static_inference():
path = './AnkleSeg.engine'
engine = load_engine(path)
imgpath = r'D:\Dataset\AnkleSegmentation\test\left_xxm-04-zm-L LV.nii.gz'
context = engine.create_execution_context(
版权声明:本文为[juluwangriyue]所创,转载请带上原文链接,感谢。 https://gsmany.com/2022/174/202206231056133536.html