Update run_deca.py

mks0601 · Dec 15, 2024 · 9033595 · 9033595
1 parent be36175
commit 9033595
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/fitting/tools/code_to_copy/run_deca.py b/fitting/tools/code_to_copy/run_deca.py
@@ -19,6 +19,7 @@ def parse_args():
 root_path = args.root_path
 
 # run DECA
+assert osp.isfile('./data/deca_model.tar'), 'please download deca_model.tar with fetch_data.sh'
 output_save_path = './flame_parmas_out'
 os.system('rm -rf ' + output_save_path)
 os.makedirs(output_save_path, exist_ok=True)