Browse Source
changed the operator
Signed-off-by: krishna katyal <krishna.katyal@zilliz.com>
main
krishna katyal
3 years ago
1 changed files with
1 additions and
1 deletions
-
deepface.py
|
|
@ -19,7 +19,7 @@ class DeepfaceFaceEmbedding(NNOperator): |
|
|
|
self.model_name=model_name |
|
|
|
@arg(1, to_image_color('RGB')) |
|
|
|
def __call__(self,img: Image) -> np.ndarray: |
|
|
|
open_cv_image = np.array(Image) |
|
|
|
open_cv_image = np.array(img) |
|
|
|
# Convert RGB to BGR |
|
|
|
open_cv_image = open_cv_image[:, :, ::-1].copy() |
|
|
|
# img.to_ndarray() |
|
|
|