From 66de165c87c18775d0201fc2122f6c6afbb0d646 Mon Sep 17 00:00:00 2001
From: krishna katyal <krishna.katyal@zilliz.com>
Date: Wed, 20 Apr 2022 22:43:32 +0530
Subject: [PATCH] Init the repository

Signed-off-by: krishna katyal <krishna.katyal@zilliz.com>
---
 deepface.py      | 33 +++++++++++++++++++++++++++++++++
 requirements.txt |  1 +
 2 files changed, 34 insertions(+)
 create mode 100644 deepface.py
 create mode 100644 requirements.txt

diff --git a/deepface.py b/deepface.py
new file mode 100644
index 0000000..2367725
--- /dev/null
+++ b/deepface.py
@@ -0,0 +1,33 @@
+from typing import NamedTuple
+import warnings
+warnings.filterwarnings("ignore")
+from towhee.utils.ndarray_utils import to_ndarray
+from deepface import DeepFace
+import numpy as np
+from towhee.types.image import Image
+from towhee.operator import NNOperator
+from towhee import register
+from towhee.types.image_utils import to_pil
+from towhee._types import Image
+from towhee.types import arg, to_image_color
+
+
+@register(output_schema=['vec'])
+class DeepfaceFaceEmbedding(NNOperator):
+  def __init__(self, model_name: str) -> None:
+      super().__init__()
+      self.model_name=model_name
+  @arg(1, to_image_color('RGB'))
+  def __call__(self,img: Image) -> np.ndarray:
+      open_cv_image = np.array(pil_image) 
+      # Convert RGB to BGR 
+      open_cv_image = open_cv_image[:, :, ::-1].copy() 
+      # img.to_ndarray()
+      embedding=DeepFace.represent(open_cv_image, model_name = self.model_name)
+      embedding=np.array(embedding)
+      return embedding
+  def train(self):
+          """
+          For training model
+          """
+          pass
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..c583e92
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1 @@
+deepface
\ No newline at end of file