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

信心