Toybrick

测试RK3568 - 不能在RKNN2上直接加载已转换好的模式

jasonzhu12

新手上路

积分
26
楼主
发表于 2021-10-10 08:04:20    查看: 3650|回复: 2 | [复制链接]    打印 | 只看该作者
错误信息:
--> Running model
Traceback (most recent call last):
  File "test_openpose.py", line 93, in <module>
    [output] = rknn.inference(inputs=[frameinput], data_format="nchw")
  File "/usr/local/lib/python3.6/dist-packages/rknn/api/rknn.py", line 350, in inference
    inputs_pass_through=inputs_pass_through, get_frame_id=get_frame_id)
  File "rknn/api/rknn_base.py", line 2539, in rknn.api.rknn_base.RKNNBase.inference
  File "rknn/api/rknn_base.py", line 2376, in rknn.api.rknn_base.RKNNBase._inference_by_compiler
TypeError: list indices must be integers or slices, not str


源代码:
    # Create RKNN object
    rknn = RKNN()

    # pre-process config
    print('--> config model')
    rknn.config(mean_values=[128, 128, 128], std_values=[128, 128, 128])
    print('done')

    # Load openpose model
    print('--> Loading model')
    # ret = rknn.load_caffe(model='./pose/coco/pose_deploy_linevec.prototxt',
    #                      proto='caffe',
    #                      blobs='./pose/coco/pose_deploy_linevec.caffemodel')
    ret = rknn.load_rknn('./openpose_v2.rknn')


...
...
...

    print('--> Init runtime environment')
    ret = rknn.init_runtime()
    ret = 0
    if ret != 0:
        print('Init runtime environment failed')
        exit(ret)
    print('done')
    t = time.time()
    frameinput = np.transpose(img, [2, 0, 1])
    # Inference
    print('--> Running model')
    [output] = rknn.inference(inputs=[frameinput], data_format="nchw")
    # show_outputs(outputs)
    elapsed = time.time() - t
    print('inference image: %.4f seconds.' % (elapsed))


###########
备注:
如果是按照例子,顺序转换原Caffe模式后,去调用rknn.inference 是能成功执行程序的。

###########
问题:
为什么不能在RKNN2上直接加载已转换好的模式 (openpose_v2.rknn)
回复

使用道具 举报

bob286

中级会员

积分
327
沙发
发表于 2021-10-11 16:51:22 | 只看该作者
从文档可以看出来,这种调用方式是不支持的,建议读下文档3.2节。
回复

使用道具 举报

jasonzhu12

新手上路

积分
26
板凳
 楼主| 发表于 2021-10-11 22:32:33 | 只看该作者
多谢了,bob286。看样子,这RKNN-Toolkit2还是有很多不支持的场景。不知道什么时间能有下个改进的版本。
回复

使用道具 举报

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

本版积分规则

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


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