Toybrick

楼主: jefferyzhang

rknn-toolkit & rknn-api for Toybrick

 

marshorizon

注册会员

积分
71
楼主
发表于 2021-3-2 10:16:17 | 显示全部楼层
你好,keras模型转rknn失败,劳驾指点一下。
环境:Ubuntu x86 toolkit 1.6,  keras 2.2.4,python 3.6
load h5模型报错:

*************************
all device(s) with ntb mode:
TM018084201100261
*************************
([], ['TM018084201100261'])
**************************************************
E Catch exception when loading keras model: ./12net.h5!
E Traceback (most recent call last):
E   File "rknn/api/rknn_base.py", line 302, in rknn.api.rknn_base.RKNNBase.load_keras
E   File "rknn/base/RKNNlib/RK_nn.py", line 233, in rknn.base.RKNNlib.RK_nn.RKnn.load_keras
E   File "rknn/base/RKNNlib/app/importer/import_keras.py", line 202, in rknn.base.RKNNlib.app.importer.import_keras.ImportKeras.run
E   File "rknn/base/RKNNlib/app/importer/import_keras.py", line 154, in rknn.base.RKNNlib.app.importer.import_keras.ImportKeras.convert_engine_keras
E   File "rknn/base/RKNNlib/converter/convert_keras.py", line 23, in rknn.base.RKNNlib.converter.convert_keras.convert_keras.__init__
E   File "/home/rafe/work/venv/lib/python3.6/site-packages/tensorflow/python/keras/_impl/keras/engine/saving.py", line 239, in load_model
E     raise ValueError('No model found in config file.')
E ValueError: No model found in config file.
-1
回复

使用道具 举报

marshorizon

注册会员

积分
71
沙发
发表于 2021-3-2 10:39:29 | 显示全部楼层
pytorch模型转换报错
在转换openpose的模型时也报错
环境:Ubuntu 18.04  x86,torch 1.6,toolkit 1.6,python 3.6,npu:toybrick 1808,报错信息:
*************************
all device(s) with ntb mode:
TM018084201100261
*************************
([], ['TM018084201100261'])
**************************************************
./kinetics-st_gcn.pt ********************
E Catch exception when loading pytorch model: ./kinetics-st_gcn.pt!
E Traceback (most recent call last):
E   File "rknn/api/rknn_base.py", line 339, in rknn.api.rknn_base.RKNNBase.load_pytorch
E   File "rknn/base/RKNNlib/RK_nn.py", line 146, in rknn.base.RKNNlib.RK_nn.RKnn.load_pytorch
E   File "rknn/base/RKNNlib/app/importer/import_pytorch.py", line 128, in rknn.base.RKNNlib.app.importer.import_pytorch.ImportPytorch.run
E   File "rknn/base/RKNNlib/converter/convert_pytorch_new.py", line 2243, in rknn.base.RKNNlib.converter.convert_pytorch_new.convert_pytorch.load
E   File ".../lib/python3.6/site-packages/torch/jit/__init__.py", line 275, in load
E     cpp_module = torch._C.import_ir_module(cu, f, map_location, _extra_files)
E RuntimeError: [enforce fail at inline_container.cc:144] . PytorchStreamReader failed reading zip archive: failed finding central directory
-1
劳驾指点一下

回复

使用道具 举报

marshorizon

注册会员

积分
71
板凳
发表于 2021-3-25 11:29:15 | 显示全部楼层
对build时dataset.txt设置的疑惑。如果do_quantization=True,dataset.txt里面为空,则模型并不能正确分类,如果do_quantization=True,dataset.txt里面为推理图片,则推理有的图片正确有的图片错误。请问dataset.txt的设置有什么要求,是训练时的train\test\validate还是推理图片?如果是推理图片,就成了鸡蛋悖论了。
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表