CoreML为对象的搜索
问题描述:
有很多教程如何打印顶部检测对象与ML的图像,就像CoreML为对象的搜索
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")
}
,但有什么办法可以解决这个问题正好相反:与给定的键字得到的概率该图片上是否存在具有此密钥的对象?
答
我不知道有什么方法可以直接使用CoreML库来获取某个对象的概率,但是您可以使用结果数组获得您感兴趣的任何概率。而不是采取results.first
,关键你有兴趣:
let interestingResults = results.filter { $0.identifier == "key" }
let topResult = interestingResults.first
,然后你可以用topResult.confidence