Pet Segmentation iOS View Controller with Fritz (www.fritz.ai)

ViewController.swift

import UIKit
import AVFoundation
import Fritz

class ViewController: UIViewController, UIImagePickerControllerDelegate,
UINavigationControllerDelegate {
    
  @IBOutlet var imageView: UIImageView!
  var maskView: UIImageView!
  var backgroundView: UIImageView!

  let context = CIContext()

  override func viewDidLoad() {
    super.viewDidLoad()
    openPhotoLibrary()
    backgroundView = UIImageView(frame: view.bounds)
    backgroundView.backgroundColor = .red
    view.addSubview(backgroundView)

    imageView = UIImageView(frame: view.bounds)
    imageView.contentMode = .scaleAspectFit
    view.addSubview(imageView)
  }

  override func viewDidAppear(_ animated: Bool) {
    super.viewDidAppear(animated)
  }
  
  func openPhotoLibrary() {
    if UIImagePickerController.isSourceTypeAvailable(.photoLibrary) {
      let imagePicker = UIImagePickerController()
      imagePicker.delegate = self
      imagePicker.sourceType = .photoLibrary
      imagePicker.allowsEditing = true
      self.present(imagePicker, animated: true, completion: nil)
    }
  }
  
  @objc func imagePickerController(_ picker: UIImagePickerController!, didFinishPickingImage image: UIImage!, editingInfo: NSDictionary!){
    self.dismiss(animated: true, completion: { () -> Void in
    })
    self.imageView.image = image
  }
 
}