From ae4f1d434d1b3faf9eaa05c43598195c94c226a1 Mon Sep 17 00:00:00 2001
From: rdanecek <danekradek@gmail.com>
Date: Wed, 20 Apr 2022 18:10:44 +0200
Subject: [PATCH] Add DECA model to the model collection
---
gdl_apps/EMOCA/demos/download_assets.sh | 13 ++++++++++++-
gdl_apps/EMOCA/utils/package_model.py | 8 +++++++-
2 files changed, 19 insertions(+), 2 deletions(-)
diff --git a/gdl_apps/EMOCA/demos/download_assets.sh b/gdl_apps/EMOCA/demos/download_assets.sh
index 97440a20..62b76438 100755
--- a/gdl_apps/EMOCA/demos/download_assets.sh
+++ b/gdl_apps/EMOCA/demos/download_assets.sh
@@ -36,10 +36,21 @@ echo "Extracting EMOCA..."
unzip EMOCA.zip
cd ../../
+echo "Downloading DECA..."
+mkdir -p EMOCA/models
+cd EMOCA/models
+wget https://download.is.tue.mpg.de/emoca/assets/EMOCA/models/DECA.zip -O DECA.zip
+echo "Extracting DECA..."
+unzip DECA.zip
+cd ../../
+
+
echo "Downloading DECA related assets"
wget https://download.is.tue.mpg.de/emoca/assets/DECA.zip -O DECA.zip
-echo "Extracting DECA..."
+wget https://download.is.tue.mpg.de/emoca/assets/FaceRecognition.zip -O FaceRecognition.zip
+echo "Extracting DECA related assets..."
unzip DECA.zip
+unzip FaceRecognition.zip
echo "Downloading FLAME related assets"
wget https://download.is.tue.mpg.de/emoca/assets/FLAME.zip -O FLAME.zip
diff --git a/gdl_apps/EMOCA/utils/package_model.py b/gdl_apps/EMOCA/utils/package_model.py
index 47333357..0160665d 100644
--- a/gdl_apps/EMOCA/utils/package_model.py
+++ b/gdl_apps/EMOCA/utils/package_model.py
@@ -118,11 +118,17 @@ def main():
if len(sys.argv) < 4:
# print("Usage: package_model.py <model_dir> <output_packaged_model_dir>")
# sys.exit()
+ # EMOA
input_dir = "/ps/project/EmotionalFacialAnimation/emoca/face_reconstruction_models/new_affectnet_split/final_models" \
"/2021_11_13_03-43-40_4753326650554236352_ExpDECA_Affec_clone_NoRing_EmoC_F2_DeSeggt_BlackC_Aug_early"
output_dir = "/ps/project/EmotionalFacialAnimation/emoca/face_reconstruction_models/new_affectnet_split/final_models/packaged2/EMOCA"
- asset_dir = "/home/rdanecek/Workspace/Repos/gdl/assets2/"
+ # DECA
+ # input_dir = "/is/cluster/work/rdanecek/emoca/finetune_deca/2021_11_15_17-20-34_-4851631063966731039_Orig_DECA2"
+ # output_dir = "/ps/project/EmotionalFacialAnimation/emoca/face_reconstruction_models/new_affectnet_split/final_models/packaged2/DECA"
+
+
+ asset_dir = "/home/rdanecek/Workspace/Repos/gdl/assets/"
# input_dir = sys.argv[1]
--
GitLab