Toybrick

npu模型推理时内存泄漏

入门小白

中级会员

积分
252
发表于 2021-4-12 15:30:29    查看: 9496|回复: 2 | [复制链接]    打印 | 显示全部楼层
本帖最后由 入门小白 于 2021-4-13 11:07 编辑

1、将mpp解码后的摄像头数据不断的进行模型推理(自己转换的模型,无后处理操作),内存会缓慢上涨(一小时大概20M);2、测试rockx中人脸检测,内存也会缓慢增长;运行你们官方的yolov5s_relu_rk180x_out_opt.rknn模型,内存上涨的更快;
3、但是推理官方的mobilenet_v1.rknn模型,内存基本没什么变化;
上述测试均只执行模型推理(python 接口),而且单独播放解码后的视频,内存是正常的,rk3399pro,DRV和API版本均为1.6.0
这是什么原因?和模型有关?


回复

使用道具 举报

我是大表书

注册会员

积分
198
发表于 2021-6-8 14:30:28 | 显示全部楼层
你解决了么,内存泄露的话你看一下ret = rknn_init(&ctx,sill_model,sill_model_len,RKNN_FLAG_PRIOR_MEDIUM);  最后一个参数是怎么设置的
回复

使用道具 举报

叽里咕噜象

注册会员

积分
90
发表于 2021-12-28 14:59:16 | 显示全部楼层
我是大表书 发表于 2021-6-8 14:30
你解决了么,内存泄露的话你看一下ret = rknn_init(&ctx,sill_model,sill_model_len,RKNN_FLAG_PRIOR_MEDIU ...

请问,最后一个参数应该设置成什么才能防止内存缓慢增加?
回复

使用道具 举报

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

本版积分规则

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


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