當前位置:網站首頁>Tensorrt筆記(四)推理分割模型

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

2022-06-23 11:02:33juluwangriyue

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://cht.chowdera.com/2022/174/202206231056133536.html

隨機推薦