Model Output Params

ModelOutputParams.swift

class ViewController: UIViewController {

  @IBOutlet var imageView: UIImageView!
  var backgroundView: UIImageView!
  var backgroundViewDelayed: UIImageView!

  let context = CIContext()

  // ------------------
  // Step 1
  // ------------------
  /// Scores output from model greater than this value will be set as 1.
  /// Lowering this value will make the mask more intense for lower confidence values.
  var clippingScoresAbove: Double { return 0.5 }

  /// Values lower than this value will not appear in the mask.
  var zeroingScoresBelow: Double { return 0.3 }

  /// Controls the opacity the mask is applied to the base image.
  var opacity: CGFloat { return 1.0 }

  /// ...
}