diff --git a/.gitignore b/.gitignore index ea2740997c121e44d879e35b030b728e7e0ad92e..681723a20ebd8775bb4a701ad660c9e6aca486ed 100644 --- a/.gitignore +++ b/.gitignore @@ -2,4 +2,5 @@ __pycache__/ .ipynb_checkpoints .vscode/ submitted.sql3 -.nfs* \ No newline at end of file +.nfs* +_cudacache/ \ No newline at end of file diff --git a/gen_latents.py b/gen_latents.py index 93cbb1e695d292c797a7e45ed878d554bff950df..3c3c8f1c9f7212181b1dbd6044a80872695e0465 100644 --- a/gen_latents.py +++ b/gen_latents.py @@ -14,7 +14,7 @@ import bob.io.base import modules import utils as sg_utils import argparse -from gridtk.tools import get_array_job_slice +# from gridtk.tools import get_array_job_slice def parse_arguments(): '''Parses in CLI arguments''' @@ -69,7 +69,7 @@ def main(): # Iterate through a single image - we use SGE_TASK_ID to parallelize list_images = sorted(os.listdir(SRC_DIR)) - list_images = list_images[get_array_job_slice(len(list_images))] + # list_images = list_images[get_array_job_slice(len(list_images))] # Instantiate the three main modules cropper, projector = instantiate_sg2_modules() diff --git a/src/facemorpher/locator.py b/src/facemorpher/locator.py index b13344f72eeb13502a51775d7b594bd1c0a7020d..cc69fe40bc1c05ab7814d123c5dae0c1adffb04a 100644 --- a/src/facemorpher/locator.py +++ b/src/facemorpher/locator.py @@ -2,6 +2,8 @@ # To view the source repository of this code, visit: # https://github.com/alyssaq/face_morpher/blob/dlib/facemorpher/locator.py +# Adjustements made by Eklavya Sarkar (Idiap Research Institute, Biometrics Security and Privacy), Jan-Feb 2021, +# Save used DLIB model previously downloaded -> DLIB_LMD_PATH = rc['sg2_morph.dlib_lmd_path'] """ Locate face points @@ -12,12 +14,13 @@ import numpy as np import os.path as path import dlib import os +from bob.extension import rc dlib_detector = dlib.get_frontal_face_detector() # The following line has been modified to remove the DLIB_DATA_DIR -dlib_predictor = dlib.shape_predictor( - '/idiap/home/esarkar/temp/Experiments/FaceMorph/bob/project/morph/gen/morphs/shape_predictor_68_face_landmarks.dat') +DLIB_LMD_PATH = rc['sg2_morph.dlib_lmd_path'] +dlib_predictor = dlib.shape_predictor(DLIB_LMD_PATH) def boundary_points(points, width_percent=0.1, height_percent=0.1): """ Produce additional boundary points