![](https://img-blog.csdnimg.cn/20201014180756780.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
文章平均质量分 55
老文化沙漠
这个作者很懒,什么都没留下…
展开
-
模型转换:pth转onnx
文章目录前言一、pth转onnx模型二、使用步骤1.引入库2.读入数据总结前言 休闲大半个月,终于又要开始认真工作了,果然游戏害人(淦),废话不多说,现在开始工作,我在转模型的时候遇到总总问题,先是转化不成功(前面有文章说过这个问题),后面转成功了但是我前几天想用那个模型的时候发现,转化的模型有大问题,首先是数值上和pth文件不一致,其次甚至没有BatchNormalization模块,具体看下图(netron查看onnx模型) 我思索了良久,下面给出转模型的全过程以及解决上述问题的方法。一原创 2021-08-26 10:09:15 · 15485 阅读 · 14 评论 -
.pth转.onnx报错:RuntimeError: Only tuples, lists and Variables supported as JIT inputs, but got dict
项目场景:最近在弄一个项目,用到了CenterNet网络(主干网络选择的是resnet,非DCN模块),转化模型一直出错:经过多次试验,是pytorch版本太低导致的,开始用的pytorch1.2一直出错,后面用pytorch1.9后成功转化模型。后续,会更新详细的转换模型中需要注意的事项,以及opencv的dnn模块加载.onnx模型进行模型推理的整体过程。...原创 2021-07-24 21:13:10 · 3355 阅读 · 2 评论