rknn-toolkit icon indicating copy to clipboard operation
rknn-toolkit copied to clipboard

nanodet-plus的torchscript模型转rknn报错`E AttributeError: 'numpy.ndarray' object has no attribute 'url'`

Open lx-r opened this issue 3 years ago • 4 comments

E Catch exception when loading pytorch model: nanodet.torchscript.pt!
E Traceback (most recent call last):
E   File "rknn/base/RKNNlib/converter/convert_pytorch_new.py", line 3985, in rknn.base.RKNNlib.converter.convert_pytorch_new.convert_pytorch.load
E   File "rknn/base/RKNNlib/converter/convert_pytorch_new.py", line 4156, in rknn.base.RKNNlib.converter.convert_pytorch_new.convert_pytorch.parse_nets
E   File "rknn/base/RKNNlib/converter/convert_pytorch_new.py", line 3783, in rknn.base.RKNNlib.converter.convert_pytorch_new.PyTorchOpConverter.convert_operators
E   File "rknn/base/RKNNlib/converter/convert_pytorch_new.py", line 2485, in rknn.base.RKNNlib.converter.convert_pytorch_new.PyTorchOpConverter.size
E   File "rknn/base/RKNNlib/RKNNnet.py", line 582, in rknn.base.RKNNlib.RKNNnet.RKNNNet.get_layer_by_out_tensor
E AttributeError: 'numpy.ndarray' object has no attribute 'url'
E Please feedback the detailed log file <log_feedback_to_the_rknn_toolkit_dev_team.log> to the RKNN Toolkit development team.
E You can also check github issues: https://github.com/rockchip-linux/rknn-toolkit/issues

lx-r avatar Jan 07 '22 02:01 lx-r

Hello, Could you upload the test model and we can take a close look on how it happen? Thank you.

zen-xingle avatar Jan 10 '22 01:01 zen-xingle

@lx-r 我也遇到了相同的问题,请问您的问题解决了吗?

mkwml avatar Jan 29 '23 01:01 mkwml

尝试了一下几种方案,最后问题解决。 1、升级pytorch到1.9.0和torchvision,再测试,load_pytorch函数可以执行成功,但是后面出现其它错误; 2、使用onnx转换模型,可以转换成功,但是预测结果错误; 3、降低onnx opset版本到10,模型转换成功,结果预测正确。

mkwml avatar Feb 06 '23 02:02 mkwml

@mkwml 想知道,您转到rknn快了得到了多少的加速呢。手上的项目对延迟的容忍很低,又想通过npu来扩大点尺寸提高准确率。

zhaodongyang2016 avatar Feb 21 '24 08:02 zhaodongyang2016