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