前言

笔者在做毕业设计时,使用到了OpenCV的DNN模块进行图像识别。而今天将项目部署到服务器上后发现了一个问题,在识别两到三个图片后进程就会被杀死。

问题排查

首先笔者查看了日志文件,确定是由于OOM导致的进程被系统杀死。

cat /var/log/message

message.png

接着,笔者又使用使用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");
        }
    }
版权声明: 本博客所有文章除特别声明外,均采用署名-非商业性使用-相同方式共享 4.0 国际CC BY-NC-SA 4.0 )许可协议。转载请注明出处!
最后修改:2021 年 07 月 18 日 10 : 32 PM
如果觉得我的文章对你有用,请随意赞赏