Toybrick

调用rknn的eval_memory接口出错

jfangah

中级会员

积分
214
楼主
发表于 2019-12-26 20:09:23    查看: 5622|回复: 2 | [复制链接]    打印 | 只看该作者
错误信息:
-->loading model
loading model done
--> Init runtime environment
W Flag eval_mem has been set, it will affect the performance of inference!
done
W RKNNAPI: IOCTL_CMD_MEMORY_DETAIL fail, ack=0, data_len=0! Please confirm that it is called after rknn_outputs_get!
E Get memory detail success, but detail is empty!
T NoneType
None


测试代码:
def load_model():
        rknn = RKNN()
        print('-->loading model')
        #rknn.load_rknn('./yolov3_tiny.rknn')
        rknn.load_rknn('./rknn_models/car_people_u8_pre.rknn')
        print('loading model done')

        print('--> Init runtime environment')
        ret = rknn.init_runtime(target = 'rk1808', device_id = 'TS018082190800200', eval_mem=True)
        if ret != 0:
                print('Init runtime environment failed')
                exit(ret)
        print('done')
        return rknn


if __name__ == '__main__':
    rknn = load_model()
    im_file = './dog.jpg'
    im = Image.open(im_file)
    im = im.resize((416, 416))
    mat = np.asarray(im.convert('RGB'))
    out_boxes, out_boxes2, out_boxes3 = rknn.inference(inputs=[mat])

    memory_detail = rknn.eval_memory(is_print=True)
    print(memory_detail)


请问这是哪里出了问题,没有内存使用的信息

回复

使用道具 举报

jefferyzhang

版主

积分
12952
沙发
发表于 2019-12-27 08:59:59 | 只看该作者
请问你运行环境是什么?
我这里问过,计算棒之前版本固件这个接口有问题,后来版本已经修复了,建议你烧写下新版本固件试试
回复

使用道具 举报

jfangah

中级会员

积分
214
板凳
 楼主| 发表于 2019-12-27 10:49:29 | 只看该作者
jefferyzhang 发表于 2019-12-27 08:59
请问你运行环境是什么?
我这里问过,计算棒之前版本固件这个接口有问题,后来版本已经修复了,建议你烧写 ...

好的,重新烧写固件解决了问题,谢谢
回复

使用道具 举报

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

本版积分规则

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


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