ViewController.swift

class ViewController: UIViewController {

    var cameraView: UIImageView!
    var maskView: UIImageView!

    override func viewDidLoad() {
        // ...
        
        cameraView = UIImageView(frame: view.bounds)
        cameraView.contentMode = .scaleAspectFill

        maskView = UIImageView(frame: view.bounds)
        maskView.contentMode = .scaleAspectFill

        cameraView.mask = maskView
        
        // ...
}

extension ViewController: AVCaptureVideoDataOutputSampleBufferDelegate {

    func captureOutput(_ output: AVCaptureOutput, didOutput sampleBuffer: CMSampleBuffer, from connection: AVCaptureConnection) {
        let image = FritzVisionImage(buffer: sampleBuffer)
        let options = FritzVisionSegmentationModelOptions(cropAndScaleOption: .scaleFit)

        visionModel.predict(image, options: options) { [weak self] (mask, error) in
            guard let mask = mask, let imageBuffer = CMSampleBufferGetImageBuffer(sampleBuffer)  else { return }

            DispatchQueue.main.async {
                self?.cameraView.image = UIImage(pixelBuffer: imageBuffer)
                self?.maskView.image = mask.toImageMask(of: FritzVisionPeopleClass.person, threshold: 0.7, minThresholdAccepted: 0.25)
            }
        }
    }
}