转化pytorch模型时报错

[复制链接] · 发表于 2020-7-24 18:53:04

pytorch 版本1.2.0+cpu
转化模型时遇到报错
E Catch exception when loading pytorch model: C:\Users\Administrator\Desktop\rknn-toolkit-master\fire_smoke\fire_smoke_1.pt!
Load pytorch model failed!
E Traceback (most recent call last):
E File "rknn\api\rknn_base.py", line 611, in rknn.api.rknn_base.RKNNBase.load_pytorch
E File "rknn\base\RKNNlib\app\importer\import_pytorch.py", line 97, in rknn.base.RKNNlib.app.importer.import_pytorch.ImportPytorch.run
E File "rknn\base\RKNNlib\converter\convert_pytorch.py", line 552, in rknn.base.RKNNlib.converter.convert_pytorch.convert_pytorch.__init__
E File "C:\Users\Administrator\AppData\Roaming\Python\Python36\site-packages\torch\jit\__init__.py", line 162, in load
E cpp_module = torch._C.import_ir_module(cu, f, map_location, _extra_files)
E RuntimeError: [enforce fail at ..\caffe2\serialize\inline_container.cc:137] . PytorchStreamReader failed reading zip archive: failed finding central directory
E (no backtrace available)

跑官方pytorch 的demo是可以转化成功的

但是加载我自己的pt文件就会报这个错误，pt文件是yolov3网络结构
https://github.com/ultralytics/yolov3 训练代码

网上搜不到相关的报错
请问该怎么解决这个问题

只看该作者 · 发表于 2020-8-10 13:51:31

你好，这个问题解决了吗？

只看该作者 · 发表于 2020-8-11 09:00:15

升级到最新rknn toolkit试试。
还是解决不了的情况下，可以加QQ群1025468710，里头有rknn开发人员帮忙解答。

只看该作者 · 发表于 2020-8-28 13:53:07

我也遇到相同的问题

只看该作者 · 发表于 2020-8-28 14:51:41

RKNN问题欢迎进群交流：http://t.rock-chips.com/forum.ph ... &extra=page%3D1

只看该作者 · 发表于 2020-9-11 09:48:07

楼主，问题解决了么

只看该作者 · 发表于 2020-9-11 09:48:42

lianbus 发表于 2020-8-28 13:53
我也遇到相同的问题

你好，这个问题解决了么

只看该作者 · 发表于 2020-10-15 20:06:16

楼主解决了吗