yolov5-lite-onnxruntime icon indicating copy to clipboard operation
yolov5-lite-onnxruntime copied to clipboard

c++ demo有内存泄漏风险

Open defzhangaa opened this issue 1 year ago • 1 comments

我的程序运行完后内存占到了6GB。。。 发现是c++ demo里面,YOLO class的构造函数里有内存泄漏: ort_session = new Session(env, widestr.c_str(), sessionOptions); ort_session在new完以后没有释放。

解决方法: 把class YOLO中的private: 删掉(或者把ort_session改为public),然后检测任务完成后添加一行delele YourYOLOModel.ort_session,其中YourYOLOModel是你的YOLO class实例名。

defzhangaa avatar Jun 23 '24 04:06 defzhangaa

当然,写一个析构函数释放掉也是可以的。

defzhangaa avatar Jun 24 '24 09:06 defzhangaa