diff --git a/evaluation_pipeline.py b/evaluation_pipeline.py index c9be0674006d93ddbe6334aa4677eaad1c3116dc..dbced92329f91c19135014b2c589568137053253 100644 --- a/evaluation_pipeline.py +++ b/evaluation_pipeline.py @@ -81,7 +81,7 @@ from bob.bio.invert.wrappers import get_invert_pipeline import os,sys sys.path.append(os.getcwd()) # import src -sys.path.append(args.path_eg3d_repo) # import eg3d files +sys.path.append(f"{args.path_eg3d_repo}/eg3d") # import eg3d files if args.attack='GaFaR': from transformers import GaFaR_InversionTransformer as InversionTransformer inv_transformer = InversionTransformer(checkpoint=args.checkpoint, eg3d_checkpoint=args.path_eg3d_checkpoint)