Fix misc

exadel-inc · Mar 5, 2024 · d676d0a · d676d0a
1 parent b7a851c
commit d676d0a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/embedding-calculator/src/constants.py b/embedding-calculator/src/constants.py
@@ -45,7 +45,7 @@ class ENV(Constants):
     PYTORCH_MODE = get_env_bool('PYTORCH_MODE', True)
     DEVICE = get_env('DEVICE', 'cpu')
     RECOGNITION_MODEL = get_env('RECOGNITION_MODEL', 'ir_50')
-    RECOGNITION_MODEL_PATH = get_env('RECOGNITION_MODEL_PATH', 'services/facescan/plugins/adaface/pretrained/adaface_ir50_ms1mv2.ckpt')
+    RECOGNITION_MODEL_PATH = get_env('RECOGNITION_MODEL_PATH', 'src/services/facescan/plugins/adaface/pretrained/adaface_ir50_ms1mv2.ckpt')
     DETECTOR_NAME = get_env('DETECTOR_NAME', 'mtcnn')
 
 

diff --git a/embedding-calculator/src/services/facescan/plugins/adaface/inference.py b/embedding-calculator/src/services/facescan/plugins/adaface/inference.py
@@ -1,12 +1,15 @@
 import src.services.facescan.plugins.adaface.net as net
 import torch
+import os
 from src.services.facescan.plugins.adaface.face_alignment import align
 import numpy as np
 from src.constants import ENV
 
 device = ENV.DEVICE
 model_mame = ENV.RECOGNITION_MODEL
-model_path = ENV.RECOGNITION_MODEL_PATH
+#model_path = ENV.RECOGNITION_MODEL_PATH
+model_path = os.path.join(os.getcwd(), ENV.RECOGNITION_MODEL_PATH)
+
 detector_model_name = ENV.DETECTOR_NAME
 
 adaface_models = {