用于对象搜索的 CoreML答案

【问题标题】：CoreML for object search用于对象搜索的 CoreML
【发布时间】：2018-03-28 01:34:59
【问题描述】：

有很多教程如何使用 ML 在图像上打印检测到的顶部对象，例如

let request = VNCoreMLRequest(model: model) { [weak self] request, error in
        guard let results = request.results as? [VNClassificationObservation],
            let topResult = results.first else {
                fatalError("unexpected result type from VNCoreMLRequest")
        }

但是有什么方法可以解决相反的问题：使用给定的关键字，获得具有该关键字的对象出现在图片上的概率？

【问题讨论】：

标签： swift machine-learning computer-vision coreml

【解决方案1】：

我不知道有什么方法可以直接使用 CoreML 库来获取某个对象的概率，但是您可以使用 results 数组来获取您感兴趣的任何概率。而不是采用 results.first，过滤对于您感兴趣的密钥：

let interestingResults = results.filter { $0.identifier == "key" }
let topResult = interestingResults.first

然后你可以通过topResult.confidence获得信心

【讨论】：