如何将图像语义分割模型用torchscipt转换成jit模式

如何用下面这个链接提供的ENet网络模型在pytorch环境下用torchscipt转成jit模型，并用jit模式进行推理
https://github.com/davidtvs/PyTorch-ENet
编译环境是pytorch，工具是vscode
https://github.com/phil-hoang/TorchScriptTutorial/blob/master/Export_Model.ipynb%E8%BF%99%E4%B8%AA%E6%98%AF%E6%95%99%E7%A8%8B%E4%BD%86%E6%98%AF%E6%88%91%E4%B8%8D%E6%87%82%E7%94%A8%E3%80%82

生成的模型转到这里
要在ENet网络分割的代码里加上jit模型转换，生成分割的jit模型，以jit模式跑一遍。

Pytorch框架TorchScript模型转换方法_程大海的博客-CSDN博客_torchscript模型转换为什么要使用TorchScript对模型进行转换？a)、TorchScript代码可以在它自己的解释器中调用，它本质上是一个受限的Python解释器。这个解释器不获取全局解释器锁，因此可以在同一个实例上同时处理多个请求。b)、这种格式允许我们将整个模型保存到磁盘上，并将其加载到另一个环境中，比如用Python以外的语言编写的服务器中c)、TorchScript提供了一种表示方式，我们可以在其中对代码进行编译器优化，以提供更有效的执行d)、TorchScript允许我们与许多后端/设备运行时进

https://blog.csdn.net/cxx654/article/details/115916275

https://blog.csdn.net/qq_35498696/article/details/107074137

需要导入模块: import torch [as 别名]
或者: from torch import jit [as 别名]
下面文章是用法
https://vimsky.com/examples/detail/python-method-torch.jit.html

可以参考这个

pytorch模型转torchscript_superjfhc的博客-CSDN博客_torchscript 文章目录目的方法tracescript目的将pytorch模型转化成torchscript目的就是为了可以在c++环境中调用pytorch模型。pytorch官方链接方法共有两种方法将pytorch模型转成torch script ，一种是trace，另一种是script。一版在模型内部没有控制流存在的话（if，for循环），直接用trace方法就可以了。如果模型内部存在控制流，那就需要用到script方法了。trace通过使用示例输入对模型的结构进行一次评估，并记录这些输入在模型中的变化过程

https://blog.csdn.net/weixin_41779359/article/details/109009192

共有两种方法将pytorch模型转成torch script ，一种是trace，另一种是script。一版在模型内部没有控制流存在的话（if，for循环），直接用trace方法就可以了。如果模型内部存在控制流，那就需要用到script方法了。