Update inference_manager.py
Browse files- inference_manager.py +1 -1
inference_manager.py
CHANGED
@@ -445,7 +445,7 @@ class ModelManager:
|
|
445 |
|
446 |
faceid_all_embeds = []
|
447 |
for image in images:
|
448 |
-
face = cv2.imread(image)
|
449 |
faces = self.app.get(face)
|
450 |
faceid_embed = torch.from_numpy(faces[0].normed_embedding).unsqueeze(0)
|
451 |
faceid_all_embeds.append(faceid_embed)
|
|
|
445 |
|
446 |
faceid_all_embeds = []
|
447 |
for image in images:
|
448 |
+
face = image#cv2.imread(image)
|
449 |
faces = self.app.get(face)
|
450 |
faceid_embed = torch.from_numpy(faces[0].normed_embedding).unsqueeze(0)
|
451 |
faceid_all_embeds.append(faceid_embed)
|