前言
笔者在做毕业设计时,使用到了OpenCV的DNN模块进行图像识别。而今天将项目部署到服务器上后发现了一个问题,在识别两到三个图片后进程就会被杀死。
问题排查
首先笔者查看了日志文件,确定是由于OOM导致的进程被系统杀死。
cat /var/log/message
接着,笔者又使用使用ps命令去观察该进程占用的内存,发现每进行一次图像识别,内存就会增加,两到三次后进程就会被系统杀死。
在查看了代码后,发现每次调用图像识别,OpenCV就会加载一次weights文件。于是考虑将加载文件的代码写在静态代码块中,随后测试成功。
解决方法
将OpenCV的DNN模块加载模型文件的代码写在静态代码块中。
static {
System.loadLibrary(Core.NATIVE_LIBRARY_NAME);
String cfgfilePath=null;
String weightsfilePath=null;
File cfgfile = FileUtil.createFile("static/yolov3_kwok.cfg");
File weightsfile = FileUtil.createFile("static/yolov3_kwok_17000.weights");
cfgfilePath=cfgfile.getPath();
weightsfilePath=weightsfile.getPath();
net = Dnn.readNetFromDarknet(cfgfilePath, weightsfilePath);
if (net.empty()) {
System.out.println("Reading Net error");
}
}