yolov5-lite-onnxruntime
yolov5-lite-onnxruntime copied to clipboard
c++ demo有内存泄漏风险
我的程序运行完后内存占到了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实例名。
当然,写一个析构函数释放掉也是可以的。