v-sher / face-search Goto Github PK

emb_res = app.get(rgb_arr)
res = emb_res[0].embedding

def generate_embs(img_fpaths: List[str]):
    embs_set = list()
    embs_label = list()
    for img_fpath in img_fpaths:  
        print('tratamento: ',img_fpath)                    
        # read grayscale img
        img = Image.open(os.path.join(YALE_DIR, img_fpath)) 
        img_arr = np.asarray(img)  
        
        # convert grayscale to rgb
        im = Image.fromarray((img_arr * 255).astype(np.uint8))
        rgb_arr = np.asarray(im.convert('RGB'))       
       
        # generate Insightface embedding
        emb_res = app.get(rgb_arr)
        try:
         res = emb_res[0].embedding
         # append emb to the eval set
         embs_set.append(res)          
         # append label to eval_label set
         embs_label.append(img_fpath.split("_")[0])
        except:
         print('no embedding found for this image')
    return embs_set, embs_label

evaluation_embs, evaluation_labels = filter_empty_embs(eval_set, eval_labels)
probe_embs, probe_labels = filter_empty_embs(probe_set, probe_labels)

evaluation_embs, evaluation_labels = eval_set, eval_labels
probe_embs, probe_labels = probe_set, probe_labels