.mlmodels 的枚举?

Enum for .mlmodels?

所以我目前正在从事一个项目,其中包括使用 CreateML、CoreML 和 Vision 进行图像识别。我仍在尝试数据集并使用 CreateML 改进模型,但如果我从一个模型更改为另一个模型,我必须手动更改 Xcode 在我导入 [=13] 时为我创建的变量 let model = example() =] 文件到项目中。所以我想用这些模型文件的名称做一个 tableView,所以如果我点击其中一个,它会将我带到“RecognitionVC”,将名称作为变量传递,这样我就可以实例化 class来自 rawValue(string) 的枚举,然后访问我无法实现的“模型变量”。

这是枚举的样子:

enum MLModels: String {
    case example
    case letters
    case ab
    case numbers
    case asl
}

这就是我想要实现的目标:

   func getModel() -> AnyClass {
        switch self {
        case .example:
            return example()
...
        }
    }

    var model: MLModel {
        switch self {
        case .example:
            return example()
...
        }
    }

我是这些机器学习内置框架的新手,如果有人能让我知道我做错了什么或者我该如何实现这一点,我会很高兴 grateful.I 附上一些屏幕截图。如果你想测试任何东西,我包括 link 到苹果的 public 模型:

link

当您编写 example() 时,它会创建一个包装器实例 class。这不是 MLModel。但是,它确实将 MLModel 作为 属性,因此您可以编写 return example().model 来执行您想要的操作。