本人之前Java调python脚本的业务环境是onnx模型对照片识别然后返回数据,对数据进行处理存储。原本是用模型识别单张图片,图片的来源是对海康的rtsp视频流进行图片的截取存放到一个文件夹当中,文件夹是动态删除的只保留十张图片供模型识别,识别在一个while(true)循环当中,每次需要得到文件夹中的照片名字然后对模型脚本的yaml配置文件修改识别路径,这样做每次line=reader.readLine()只得到一条数据没有错误,但项目跑起来之后很吃cpu资源,运行一会之后电脑直接卡死鼠标都动不了,后来尝试用GPU去对模型识别进行处理以缓解cpu的负荷,采用GPU之后同样也是电脑直接卡死,因为gpu的内存直接满了。
直接说原因吧,在循环当中调取模型每一次都要去加载模型文件消耗大量资源,所以说不能用识别单张图片的方式去进行模型识别,要让模型去识别视频流,在py文件中修改识别速度即可,这样无论采用cpu还是gpu识别都不会太消耗资源。
视频流识别每一秒返回一次数据就会出现阻塞,Java不能及时接收到输出流,解决办法就是在调用脚本的命令中间加一个 -u,比如
String[] command = new String[]{"python", "-u", "detect1.py"};
这样就能解决问题啦!