logo
Browse Source

changed the operator

Signed-off-by: krishna katyal <krishna.katyal@zilliz.com>
main
krishna katyal 3 years ago
parent
commit
bfc93a57de
  1. 2
      deepface.py

2
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()

Loading…
Cancel
Save