在使用地平线 Python SDK (pyeasy_dnn) 运行 YOLOv5 模型时,
关键代码:
from hobot_dnn import pyeasy_dnn as dnn
models = dnn.load('xxx.bin') results = models[0].forward(nv12_data) 在使用地平线 Python SDK (pyeasy_dnn) 运行 YOLOv5 模型时,调用 models[0].forward(nv12_data) 推理图像,每帧执行一次,程序会持续增长内存,最终导致 OOM。
已确认情况: 每一帧都会调用 forward() 方法获取推理结果; SDK 中未找到的资源释放函数; 使用 del outputs 并不会阻止内存持续增长; 确认内存泄漏位于 BPU 推理输出阶段。
请求: 是否有推荐的方式在 Python 中正确释放 forward() 结果返回的资源?是否 SDK 有新版本提供了释放函数或上下文管理机制?


