Toybrick

标题: npu模型推理时内存泄漏 [打印本页]

作者: 入门小白 时间: 2021-4-12 15:30
标题: npu模型推理时内存泄漏
本帖最后由入门小白于 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
这是什么原因？和模型有关？

作者: 我是大表书 时间: 2021-6-8 14:30
你解决了么，内存泄露的话你看一下ret = rknn_init(&ctx,sill_model,sill_model_len,RKNN_FLAG_PRIOR_MEDIUM); 最后一个参数是怎么设置的

作者: 叽里咕噜象 时间: 2021-12-28 14:59

我是大表书发表于 2021-6-8 14:30
你解决了么，内存泄露的话你看一下ret = rknn_init(&ctx,sill_model,sill_model_len,RKNN_FLAG_PRIOR_MEDIU ...

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

欢迎光临 Toybrick (https://t.rock-chips.com/)