Swift 5 UIImage white background removal

UIImage+removeBackground.swift

extension UIImage {

    func removeBackground() -> UIImage? {

        if let rawImageRef = self.cgImage {

            let colorMasking: [CGFloat] = [200, 255, 200, 255, 200, 255]

            UIGraphicsBeginImageContext(self.size)

            if let maskedImageRef = rawImageRef.copy(maskingColorComponents: colorMasking) {

                UIGraphicsGetCurrentContext()!.translateBy(x: 0.0, y: self.size.height)
                UIGraphicsGetCurrentContext()!.scaleBy(x: 1.0, y: -1.0)

                UIGraphicsGetCurrentContext()!.draw(maskedImageRef, in: CGRect(x: 0, y: 0, width: self.size.width, height: self.size.height))

                let result = UIGraphicsGetImageFromCurrentImageContext()

                UIGraphicsEndImageContext()

                return result
            }
        }

        return nil
    }
}