CoreML pour la recherche d'objets

Question

Il y a de nombreux tutoriels comment imprimer haut de découverte d'objets sur l'image avec ML, comme

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") 
     } 

, mais est-il un moyen de résoudre le problème inverse: avec un mot-clé donné obtenir la probabilité de cela l'objet avec cette clé est présent sur l'image?

Answer 1

Je ne connais aucun moyen d'utiliser la bibliothèque CoreML directement pour obtenir la probabilité d'un certain objet, mais vous pouvez utiliser le tableau de résultats pour obtenir toute probabilité qui vous intéresse. Au lieu de prendre results.first, filtrez pour la clé qui vous intéresse:

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

et vous pouvez obtenir la confiance avec topResult.confidence