Есть ли способ обнаружения объектов в ios без классификации объектов? - PullRequest
Новая
       12

Есть ли способ обнаружения объектов в ios без классификации объектов?

0 голосов
/ 23 января 2019

Цель - обнаружение объекта без классификации в ios.

У меня есть модель tflite для использования в xcode, но я нашел возможные способы работы в качестве классификатора.Я также пытался преобразовать модель в CoreML, но она не работает должным образом.

Ниже приведен код, который вызывается каждый раз при захвате кадра и загрузке модели: 

func captureOutput(_ output: AVCaptureOutput, didOutput sampleBuffer: CMSampleBuffer, from connection: AVCaptureConnection) {
    guard let model = try? VNCoreMLModel(for: Resnet50().model) else { return }
    let request = VNCoreMLRequest(model: model) { (finishedRequest, error) in
        guard let results = finishedRequest.results as? [VNClassificationObservation] else { return }
        guard let Observation = results.first else { return }

        DispatchQueue.main.async(execute: {
            self.label.text = "\(Observation.identifier)"
            print(Observation.confidence)
        })
    }
     guard let pixelBuffer: CVPixelBuffer = CMSampleBufferGetImageBuffer(sampleBuffer) else { return }

    // executes request
    try? VNImageRequestHandler(cvPixelBuffer: pixelBuffer, options: [:]).perform([request])
}

Может кто-нибудь помочь мне с этим?

...