runCoreImage-person-segmentation.swift

func runCoreImage(){
  let filter = CIFilter.personSegmentation()
  let input = CIImage(cgImage: inputImage.cgImage!)
  filter.inputImage = CIImage(cgImage: inputImage.cgImage!)

  if let maskImage = filter.outputImage{

      let ciContext = CIContext(options: nil)

      let maskScaleX = input.extent.width / maskImage.extent.width
      let maskScaleY = input.extent.height / maskImage.extent.height

      let maskScaled =  maskImage.transformed(by: __CGAffineTransformMake(maskScaleX, 0, 0, maskScaleY, 0, 0))

      let maskRef = ciContext.createCGImage(maskScaled, from: maskScaled.extent)
      self.outputImage = UIImage(cgImage: maskRef!)

  }
}