在不对对象进行分类的情况下,有什么方法可以在ios中进行对象检测?

时间:2019-01-23 06:19:08

标签: ios swift xcode tensorflow coreml

目的-在不进行ios分类的情况下检测对象。

我有一个在xcode中使用的tflite模型,但是我发现的可能方法正在用作分类器。我也尝试在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])
}

有人可以帮我吗?

0 个答案:

没有答案