swift - 用于对象搜索的 CoreML

Question

有很多教程如何使用 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")
        }

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

Answer 1

我不知道有什么方法可以直接使用 CoreML 库来获取某个对象的概率，但是您可以使用结果数组来获取您感兴趣的任何概率。而不是获取 results。首先，过滤你感兴趣的键:

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

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