如何编写CMakeLists.txt以将.mlmodel添加到xcode项目

Question

我有一个detect_model.mlmodel，它是由我自己训练的。我想在xcode中使用它，但是我不想手动将其添加到我的xcode项目中。我试图写CMakeLists.txt来管理我的xcode项目。但是，如何写CMake文件以将detect_model.mlmodel添加到我的xcode项目中并生成detect_model.h头文件？

我尝试使用set(mlmodel detect_model.mlmodel)和add_library(dst ${mlmodel})，但这是不正确的。

预期结果是我可以使用CMake将detect_model.mlmodel添加到xcode项目中并自动生成detect_model.h。

Answer 1

您可能需要添加一个自定义规则，该规则调用coremlc来生成.h abd .m文件和.mlmodelc文件夹。

这是您在终端机上执行的操作：

mkdir output
xcrun coremlc compile YourModel.mlmodel output
xcrun coremlc generate Model.mlmodel output

您还需要添加一条规则，以将.mlmodelc文件夹复制到应用包中。