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
生成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