aquamarine_的博客

原创 C#(winform)调用pytorch模型

winform调用pytorch上训练好的unet模型：项目是写一个辅助诊断系统软件，用winform写软件，调用pytorch和matlab的模型。这篇博客只包含调用pytorch模型的部分。1.c++(libtorch)调用模型2.c++生成动态链接库3.c#调用dll1. libtorch(cpu)调用gpu模型首先把pytorch的模型转成libtorch的。如果模型没有控制流（if-else语句），就用简单的trace方式进行转换。pth文件转为pt文件import torch