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

时间:2020-10-19 22:34:40

标签: python opencv deep-learning pytorch onnx

我正在尝试使用以下模型:https://github.com/zllrunning/face-makeup.PyTorch/blob/master/cp/79999_iter.pth与opencv dnn模块一起使用,但是由于opencv无法读取.pth文件,因此我需要将其转换为另一种格式。

首先,我尝试转换为.onnx,我找到了执行此操作的代码,但是当我调用函数“ torch.onnx.export”时,出现了“ RuntimeError:ONNX符号在跟踪中期望一个恒定值”。 我尝试使用不同版本的火炬,但总是收到几乎相同的错误消息。

我试图转换为caffe模型,有一些可用的github可以做到,但是没有一个对我有用。

我没有找到如何直接转换为tensorflow的方法,它们总是先转换为onnx然后再转换为tensorflow。

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

0 个答案:

没有答案