当前位置：网站首页>Tensorrt笔记（四）推理分割模型

Tensorrt笔记（四）推理分割模型

2022-06-23 11:00:00 【juluwangriyue】

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://blog.csdn.net/juluwangriyue/article/details/125416334

边栏推荐

猜你喜欢

随机推荐