ARSessionDelegate

ARSessionDelegate.swift

// MARK: - ARSessionDelegate

var currentBuffer: CVPixelBuffer?

func session(_: ARSession, didUpdate frame: ARFrame) {
    // We return early if currentBuffer is not nil or the tracking state of camera is not normal
    guard currentBuffer == nil, case .normal = frame.camera.trackingState else {
        return
    }

    // Retain the image buffer for Vision processing.
    currentBuffer = frame.capturedImage

    startDetection()
}

// MARK: - Private functions

private func startDetection() {
    // Here we will do our CoreML request on currentBuffer

    // Release currentBuffer to allow processing next frame
    currentBuffer = nil
}