From 23ed39c38b339d9f5492859016d4ea6461a45ea8 Mon Sep 17 00:00:00 2001
From: Frederik Juutilainen <frederik@tollund.cc>
Date: Mon, 26 Apr 2021 10:44:23 +0200
Subject: [PATCH] added function to get anonymized image

---
 deface/deface.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/deface/deface.py b/deface/deface.py
index ad53724..4feccbf 100755
--- a/deface/deface.py
+++ b/deface/deface.py
@@ -199,6 +199,29 @@ def get_file_type(path):
     return mime
 
 
+def get_anonymized_image(frame,
+                         threshold: float,
+                         replacewith: str,
+                         mask_scale: float,
+                         ellipse: bool,
+                         draw_scores: bool,
+                         ):
+    """
+    Method for getting an anonymized image without CLI
+    returns frame
+    """
+
+    centerface = CenterFace(in_shape=None, backend='auto')
+    dets, _ = centerface(frame, threshold=threshold)
+
+    anonymize_frame(
+        dets, frame, mask_scale=mask_scale,
+        replacewith=replacewith, ellipse=ellipse, draw_scores=draw_scores
+    )
+
+    return frame
+
+
 def parse_cli_args():
     parser = argparse.ArgumentParser(description='Video anonymization by face detection', add_help=False)
     parser.add_argument(