Add anti_spoofing to represent function

pablo-gbr · Jun 19, 2024 · 6a575a4 · 6a575a4
1 parent 38b80a6
commit 6a575a4
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 11 deletions.
diff --git a/scripts/represent.py b/scripts/represent.py
@@ -5,6 +5,7 @@
     img_path = "{{img_path}}",
     model_name = "{{model_name}}",
     enforce_detection = {{enforce_detection}},
+    anti_spoofing = "{{anti_spoofing}}",
     detector_backend = "{{detector_backend}}",
     align = {{align}},
     normalization = "{{normalization}}"

diff --git a/src/DeepFace.php b/src/DeepFace.php
@@ -289,24 +289,30 @@ public function represent(
         Detector $detector_backend = Detector::OPENCV,
         bool $align = true,
         Normalization $normalization = Normalization::BASE,
+        bool $anti_spoofing = false,
     ): array {
         $img = new SplFileInfo($img_path);
 
         if (! $img->isFile()) {
             throw new InvalidArgumentException("The path [{$img_path}] for image is not a file.");
         }
 
-        $output = $this->run(
-            filepath: __DIR__.'/../scripts/represent.py',
-            data: [
-                '{{img_path}}' => str_replace('\\', '/', $img->getRealPath()),
-                '{{model_name}}' => $model_name->value,
-                '{{enforce_detection}}' => $enforce_detection ? 'True' : 'False',
-                '{{detector_backend}}' => $detector_backend->value,
-                '{{align}}' => $align ? 'True' : 'False',
-                '{{normalization}}' => $normalization->value,
-            ],
-        );
+        try{
+            $output = $this->run(
+                filepath: __DIR__.'/../scripts/represent.py',
+                data: [
+                    '{{img_path}}' => str_replace('\\', '/', $img->getRealPath()),
+                    '{{model_name}}' => $model_name->value,
+                    '{{enforce_detection}}' => $enforce_detection ? 'True' : 'False',
+                    '{{anti_spoofing}}' => $anti_spoofing ? 'True' : 'False',
+                    '{{detector_backend}}' => $detector_backend->value,
+                    '{{align}}' => $align ? 'True' : 'False',
+                    '{{normalization}}' => $normalization->value,
+                ],
+            );
+        } catch(DeepFaceException $e){
+            return array("error" => $e->getMessage());
+        }
 
         return array_map(
             fn (array $result) => new RepresentResult(