先将caffemodel使用下边的转成.onnx
mmconvert -sf caffe -iw path_to_caffemodel -in path_to_prototxt -df onnx -om output_onnx_model.onnx
再使用转换为torch的.pth
import torch
import onnx
from onnx2pytorch import ConvertModel
# 加载ONNX模型
onnx_model = onnx.load('model.onnx')
pytorch_model = ConvertModel(onnx_model)
# 保存PyTorch模型
torch.save(onnx_model, 'dex_imdb_wiki.pth')
print('saved!')
注意,你需要找到一个onnx和mmconvert都适用的onnx配套库版本,我先装的mmconvert,然后再装onnx,装能适配的onnx的版本(onnx是老版本,我用的是1.13.0)