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.first ，过滤您感兴趣的密钥：

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

然后您可以通过topResult.confidence

获得信心