import UIkit inside Imageclassifier.

smileidentity · Aug 29, 2024 · 65a51be · 65a51be
1 parent ae0ff57
commit 65a51be
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 119 deletions.
diff --git a/Example/Podfile.lock b/Example/Podfile.lock
@@ -43,7 +43,7 @@ SPEC CHECKSUMS:
   lottie-ios: fcb5e73e17ba4c983140b7d21095c834b3087418
   netfox: 9d5cc727fe7576c4c7688a2504618a156b7d44b7
   Sentry: 96ae1dcdf01a644bc3a3b1dc279cecaf48a833fb
-  SmileID: 8d3a64e845a5fb82239516def978391d4d814c9e
+  SmileID: 97a579344f72c543d6b3aa8b56722d05646d2023
   SwiftLint: 3fe909719babe5537c552ee8181c0031392be933
   ZIPFoundation: b8c29ea7ae353b309bc810586181fd073cb3312c
 

diff --git a/Sources/SmileID/Classes/FaceDetector/FaceDetectorV2.swift b/Sources/SmileID/Classes/FaceDetector/FaceDetectorV2.swift
@@ -8,6 +8,7 @@ protocol FaceDetectorDelegate: NSObjectProtocol {
 }
 
 class FaceDetectorV2: NSObject {
+    private let cropSize = (width: 120, height: 120)
     private let faceMovementThreshold: CGFloat = 0.15
 
     var sequenceHandler = VNSequenceRequestHandler()
@@ -48,7 +49,7 @@ class FaceDetectorV2: NSObject {
             guard let croppedImage = try cropToFace(image: image) else {
                 return
             }
-            guard let convertedImage = croppedImage.pixelBuffer(width: 120, height: 120) else {
+            guard let convertedImage = croppedImage.pixelBuffer(width: cropSize.width, height: cropSize.height) else {
                 return
             }
             selfieQualityRequest(imageBuffer: convertedImage)

diff --git a/Sources/SmileID/Classes/FaceDetector/ImageClassifier.swift b/Sources/SmileID/Classes/FaceDetector/ImageClassifier.swift