将.pth Pytorch模型转换为OpenCv可读的格式

2024-05-19 14:00:45 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图将此模型:https://github.com/zllrunning/face-makeup.PyTorch/blob/master/cp/79999_iter.pth与opencv dnn模块一起使用,但由于opencv不读取.pth文件,因此需要将其转换为其他格式

首先,我试图转换为.onnx,我找到了执行此操作的代码,但当我调用函数“torch.onnx.export”时,我得到了“RuntimeError:onnx symbolic在跟踪中需要一个常量值”。 我尝试使用不同版本的torch,但总是收到几乎相同的错误消息

我试图转换到caffe模型,有一些可用的github可以做到这一点,但没有一个适合我

我没有发现如何直接转换为tensorflow,它们总是先转换为onnx,然后再转换为tensorflow

如果有人可以与我共享适用于此模型的代码以及所使用的所有软件包的版本。或者,如果你能与我分享转换成另一种格式的模型,我将不胜感激


Tags: 代码https模型版本githubcomtensorflow格式