react-native-vision-camera-spoof-detector 1.1.25 → 1.1.26

package/VisionCameraSpoofDetector.podspec

@@ -34,6 +34,7 @@ Pod::Spec.new do |s|
   s.resources = ['lib/assets/*']
 
   s.dependency 'TensorFlowLiteSwift'
+  s.dependency 'TensorFlowLiteC/Metal'
   s.dependency 'VisionCamera'
   s.dependency 'React-jsi'
   s.dependency 'React-callinvoker'

package/ios/FaceAntiSpoofManager.swift

@@ -3,6 +3,27 @@ import Foundation
 #if canImport(TensorFlowLite)
 import TensorFlowLite
 #endif
+#if canImport(TensorFlowLiteCMetal)
+import TensorFlowLiteCMetal
+#endif
+
+#if canImport(TensorFlowLite) && canImport(TensorFlowLiteCMetal)
+private final class IOSMetalDelegate: Delegate {
+    let cDelegate: Delegate.CDelegate
+
+    init() {
+        var options = TFLGpuDelegateOptions()
+        options.allow_precision_loss = false
+        options.wait_type = TFLGpuDelegateWaitTypePassive
+        options.enable_quantization = true
+        cDelegate = TFLGpuDelegateCreate(&options)
+    }
+
+    deinit {
+        TFLGpuDelegateDelete(cDelegate)
+    }
+}
+#endif
 
 @objcMembers
 public class FaceAntiSpoofManager: NSObject {
@@ -16,7 +37,16 @@ public class FaceAntiSpoofManager: NSObject {
 
     #if canImport(TensorFlowLite)
     public static var sharedInterpreter: Interpreter?
+    public static private(set) var currentAccelerator = "CPU"
+    public static var enableGpu = true
+    public static var disableGpuDelegate = false
+    public static var enableCpuFallback = true
+    public static var forceCpuOnly = false
+    public static var cpuThreadCount = 4
     private static let interpreterLock = NSLock()
+    #if canImport(TensorFlowLiteCMetal)
+    private static var metalDelegate: IOSMetalDelegate?
+    #endif
     #endif
 
     @objc public class func initializeModel() -> Bool {
@@ -46,15 +76,48 @@ public class FaceAntiSpoofManager: NSObject {
         }
 
         do {
+            let threadCount = max(1, cpuThreadCount)
             var options = Interpreter.Options()
-            options.threadCount = 2
-            sharedInterpreter = try Interpreter(modelPath: path, options: options)
+            options.threadCount = threadCount
+
+            #if canImport(TensorFlowLiteCMetal)
+            if enableGpu && !disableGpuDelegate && !forceCpuOnly {
+                do {
+                    let delegate = IOSMetalDelegate()
+                    sharedInterpreter = try Interpreter(
+                        modelPath: path,
+                        options: options,
+                        delegates: [delegate]
+                    )
+                    metalDelegate = delegate
+                    currentAccelerator = "GPU"
+                    NSLog("[FaceAntiSpoof] Manager: Metal delegate enabled | Threads: \(threadCount)")
+                } catch {
+                    metalDelegate = nil
+                    NSLog("[FaceAntiSpoof] Manager: Metal delegate unavailable: \(error)")
+                    guard enableCpuFallback else { throw error }
+                }
+            }
+            #endif
+
+            if sharedInterpreter == nil {
+                guard enableCpuFallback || forceCpuOnly else {
+                    throw NSError(domain: "FaceAntiSpoof", code: 4, userInfo: [
+                        NSLocalizedDescriptionKey: "No enabled TensorFlow Lite accelerator is available"
+                    ])
+                }
+                sharedInterpreter = try Interpreter(modelPath: path, options: options)
+                currentAccelerator = "CPU"
+                NSLog("[FaceAntiSpoof] Manager: CPU fallback enabled | Threads: \(threadCount)")
+            }
             try sharedInterpreter?.allocateTensors()
             NSLog("[FaceAntiSpoof] Manager: Loaded model at \(path)")
             return true
         } catch {
             NSLog("[FaceAntiSpoof] Manager: Failed to create interpreter: \(error)")
             sharedInterpreter = nil
+            metalDelegate = nil
+            currentAccelerator = "CPU"
             return false
         }
         #else
@@ -71,6 +134,14 @@ public class FaceAntiSpoofManager: NSObject {
         #endif
     }
 
+    @objc public class func accelerator() -> String {
+        #if canImport(TensorFlowLite)
+        return currentAccelerator
+        #else
+        return "UNAVAILABLE"
+        #endif
+    }
+
     public class func process(input: [Float]) throws -> DetectionResult {
         #if canImport(TensorFlowLite)
         if sharedInterpreter == nil && !initializeModel() {
@@ -158,7 +229,11 @@ public class FaceAntiSpoofManager: NSObject {
 
     @objc public class func cleanupModel() {
         #if canImport(TensorFlowLite)
+        interpreterLock.lock()
         sharedInterpreter = nil
+        metalDelegate = nil
+        currentAccelerator = "CPU"
+        interpreterLock.unlock()
         #endif
         NSLog("[FaceAntiSpoof] Manager: cleaned up model")
     }

package/ios/FaceAntiSpoofModule.m

@@ -110,10 +110,26 @@ RCT_REMAP_METHOD(getModuleInfo,
 								 getModuleInfoWithResolver:(RCTPromiseResolveBlock)resolve
 								 rejecter:(RCTPromiseRejectBlock)reject)
 {
-	NSDictionary *info = @{@"name": @"FaceAntiSpoof", @"methods": @[@"initialize", @"checkModelStatus", @"isAvailable", @"testMethod", @"getModuleInfo", @"install", @"cleanup"]};
+	NSDictionary *info = @{ @"name": @"FaceAntiSpoof", @"methods": @[@"initialize", @"checkModelStatus", @"isAvailable", @"testMethod", @"getModuleInfo", @"install", @"cleanup", @"getAccelerator"], @"accelerators": @[@"GPU", @"CPU"] };
 	resolve(info);
 }
 
+RCT_REMAP_METHOD(getAccelerator,
+								 getAcceleratorWithResolver:(RCTPromiseResolveBlock)resolve
+								 rejecter:(RCTPromiseRejectBlock)reject)
+{
+	Class managerClass = NSClassFromString(@"FaceAntiSpoofManager");
+	if (managerClass != Nil) {
+		SEL sel = NSSelectorFromString(@"accelerator");
+		if ([managerClass respondsToSelector:sel]) {
+			id (*msgSendObject)(id, SEL) = (id (*)(id, SEL))objc_msgSend;
+			resolve(msgSendObject((id)managerClass, sel));
+			return;
+		}
+	}
+	resolve(@"UNAVAILABLE");
+}
+
 RCT_REMAP_METHOD(install,
 								 installWithResolver:(RCTPromiseResolveBlock)resolve
 								 rejecter:(RCTPromiseRejectBlock)reject)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "react-native-vision-camera-spoof-detector",
-  "version": "1.1.25",
+  "version": "1.1.26",
   "description": "High-performance face anti-spoofing and liveness detection module for React Native Vision Camera. Uses TensorFlow Lite with GPU acceleration and optimized YUV processing.",
   "homepage": "https://github.com/dpraful/react-native-vision-camera-spoof-detector",
   "repository": {