将python sklearn模型导出到生产环境(java/c++)

Exporting python sklearn models to production (java/c++)

本文关键字:java c++ 生产环境 sklearn python 模型      更新时间:2023-10-16

我训练了一个由两个组件组成的计算机视觉分类器:数据的核PCA变换和SVM二进制分类模型。

这些模型是使用SKlearn在Python中训练的,但我想将它们用于c++和更高版本的Java中的实际计算机视觉任务。将模型导出到不同环境的最佳方式是什么?有没有一种聪明的方法可以做到这一点,或者我只需要用一种新的语言编写所有参数的代码?

我正在进行一个名为sklearn-porter的项目,该项目将经过训练的模型移植到C、Java和JavaScript等低级别编程语言中。在给定的时间将添加更多的分类器和回归器。看一看

我建议您可以直接在opencv、c++或java中完成整个过程。

这里有一个PCA和另一个SVM的例子。