Delete embeddings.py

embeddings.py

@@ -1,33 +0,0 @@
-import os
-import cv2
-import numpy as np
-import insightface
-from insightface.app import FaceAnalysis
-from glob import glob
-from tqdm import tqdm
-
-def get_embeddings(db_dir):
-    app = FaceAnalysis(name='buffalo_l')
-    app.prepare(ctx_id=0, det_size=(640, 640),)
-    names = []
-    embeddings = []
-
-    folders = os.listdir(os.path.join(db_dir))
-    for folder in tqdm(folders):
-        if ".ipynb_checkpoints" in folder:
-            continue
-        img_paths = glob(os.path.join(db_dir,  folder, '*'))
-        for img_path in img_paths:
-            img = cv2.imread(img_path)
-            if img is None:
-                continue
-            faces = app.get(img)
-            if len(faces) != 1:
-                continue
-            face = faces[0]
-            names.append(folder)
-            embeddings.append(face.normed_embedding)
-
-    embeddings = np.stack(embeddings, axis=0)
-    np.save(os.path.join(db_dir, "embeddings.npy"), embeddings)
-    np.save(os.path.join(db_dir, "names.npy"), names)