diff --git a/gdl_apps/EMOCA/demos/test_emoca_on_images.py b/gdl_apps/EMOCA/demos/test_emoca_on_images.py index 8ea9834..edebf4c 100644 --- a/gdl_apps/EMOCA/demos/test_emoca_on_images.py +++ b/gdl_apps/EMOCA/demos/test_emoca_on_images.py @@ -36,7 +36,7 @@ def main(): # add the input folder arg parser.add_argument('--input_folder', type=str, default= str(Path(gdl.__file__).parents[1] / "assets/data/EMOCA_test_example_data/images/affectnet_test_examples")) parser.add_argument('--output_folder', type=str, default="image_output", help="Output folder to save the results to.") - parser.add_argument('--model_name', type=str, default='EMOCA', help='Name of the model to use.') + parser.add_argument('--model_name', type=str, default='EMOCA_v2_lr_mse_20', help='Name of the model to use.') parser.add_argument('--path_to_models', type=str, default=str(Path(gdl.__file__).parents[1] / "assets/EMOCA/models")) parser.add_argument('--save_images', type=bool, default=True, help="If true, output images will be saved") parser.add_argument('--save_codes', type=bool, default=False, help="If true, output FLAME values for shape, expression, jaw pose will be saved") diff --git a/gdl_apps/EMOCA/demos/test_emoca_on_video.py b/gdl_apps/EMOCA/demos/test_emoca_on_video.py index e71ed04..0565f6d 100644 --- a/gdl_apps/EMOCA/demos/test_emoca_on_video.py +++ b/gdl_apps/EMOCA/demos/test_emoca_on_video.py @@ -133,7 +133,7 @@ def parse_args(): parser.add_argument('--input_video', type=str, default=str(Path(gdl.__file__).parents[1] / "/assets/data/EMOCA_test_example_data/videos/82-25-854x480_affwild2.mp4"), help="Filename of the video for reconstruction.") parser.add_argument('--output_folder', type=str, default="video_output", help="Output folder to save the results to.") - parser.add_argument('--model_name', type=str, default='EMOCA', help='Name of the model to use. Currently EMOCA or DECA are available.') + parser.add_argument('--model_name', type=str, default='EMOCA_v2_lr_mse_20', help='Name of the model to use. Currently EMOCA or DECA are available.') parser.add_argument('--path_to_models', type=str, default=str(Path(gdl.__file__).parents[1] / "assets/EMOCA/models")) parser.add_argument('--mode', type=str, default="detail", choices=["detail", "coarse"], help="Which model to use for the reconstruction.") parser.add_argument('--save_images', type=str2bool, default=True, help="If true, output images will be saved")