Toybrick

标题: 转化pytorch模型时报错 [打印本页]

作者: SodrSnne    时间: 2020-7-24 18:53
标题: 转化pytorch模型时报错
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  训练代码

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

作者: apupu    时间: 2020-8-10 13:51
你好,这个问题解决了吗?
作者: leok    时间: 2020-8-11 09:00
升级到最新rknn toolkit试试。
还是解决不了的情况下,可以加QQ群1025468710,里头有rknn开发人员帮忙解答。
作者: lianbus    时间: 2020-8-28 13:53
我也遇到相同的问题
作者: jefferyzhang    时间: 2020-8-28 14:51

RKNN问题欢迎进群交流:http://t.rock-chips.com/forum.ph ... &extra=page%3D1
作者: kkk    时间: 2020-9-11 09:48
楼主,问题解决了么
作者: kkk    时间: 2020-9-11 09:48
lianbus 发表于 2020-8-28 13:53
我也遇到相同的问题

你好,这个问题解决了么




欢迎光临 Toybrick (http://t.rock-chips.com/) Powered by Discuz! X3.3