IdentifyKit

使用 Swift 包将分类 coreML 模型轻松集成到您的代码中。

使用 CocoaPods 安装

pod 'IdentifyKit'

快速开始

首先创建一个 IdentifyKitDelegate，这将处理任何识别或识别失败的结果。

extension ViewController: IdentifyKitDelegate {
    func failedToInitialize(error: String) {
        print("Failed to initialize identifier request: \(error)")
    }
    
    
    func didIdentifyObject(name: String) {
        print("Identified: \(name)")
    }
    
    func identifying() {
        print("Identifying")
    }
    
    
    func failedToIdentifyObject() {
        print("Identification Failed")
    }
    
}

一旦设置了您的代理，您就可以初始化您的 IdentyKit 对象。初始化器接受 3 个参数

上面声明的代理。
所需的精确度，这是一个 0 到 1 之间的浮点数，将用于过滤掉低于此值的任何识别。
模型，可以是任何图像分类模型。我们在这个例子中使用了 MobileNet。

let classifier = IdentifyKit(delegate: self, accuracy: Configuration.accuracy, model: MobileNet().model)

完成这些后，您可以发出请求

func identify(image: UIImage) {
    func identify(image: UIImage) {
        let image = UIImage()
        guard let data = image.pngData() else { return }
        classifier.identify(data)
    }
}

IdentifyKit 1.21

IdentifyKit 1.21

IdentifyKit 1.21

IdentifyKit