Add Action Unit model to gpu config

config.merged.gpu.yml

@@ -19,7 +19,7 @@ analyzer:
     downloader:
       _target_: facetorch.downloader.DownloaderGDrive
       file_id: 154x2VjmTQVqmowB0yZw4Uck7uQs2vVBs
-      path_local: /code/models/torchscript/detector/1/model.pt
+      path_local: /home/user/code/models/torchscript/detector/1/model.pt
     device:
       _target_: torch.device
       type: ${analyzer.device}
@@ -102,7 +102,7 @@ analyzer:
       downloader:
         _target_: facetorch.downloader.DownloaderGDrive
         file_id: 19h3kqar1wlELAmM5hDyj9tlrUh8yjrCl
-        path_local: /code/models/torchscript/predictor/embed/1/model.pt
+        path_local: /home/user/code/models/torchscript/predictor/embed/1/model.pt
       device:
         _target_: torch.device
         type: ${analyzer.device}
@@ -142,8 +142,8 @@ analyzer:
       _target_: facetorch.analyzer.predictor.FacePredictor
       downloader:
         _target_: facetorch.downloader.DownloaderGDrive
-        file_id: 1WI-mP_0mGW31OHfriPUsuFS_usYh_W8p
-        path_local: /code/models/torchscript/predictor/verify/2/model.pt
+        file_id: 1H-aPtFd9C5D7y1vzoWsObKAxeIBE9QPd
+        path_local: /home/user/code/models/torchscript/predictor/verify/1/model.pt
       device:
         _target_: torch.device
         type: ${analyzer.device}
@@ -158,18 +158,18 @@ analyzer:
             - 112
           - _target_: torchvision.transforms.Normalize
             mean:
-            - 0.5
-            - 0.5
-            - 0.5
+            - 0.485
+            - 0.456
+            - 0.406
             std:
-            - 0.5
-            - 0.5
-            - 0.5
+            - 0.229
+            - 0.224
+            - 0.225
         device:
           _target_: torch.device
           type: ${analyzer.predictor.verify.device.type}
         optimize_transform: ${analyzer.optimize_transforms}
-        reverse_colors: true
+        reverse_colors: false
       postprocessor:
         _target_: facetorch.analyzer.predictor.post.PostEmbedder
         transform: None
@@ -184,7 +184,7 @@ analyzer:
       downloader:
         _target_: facetorch.downloader.DownloaderGDrive
         file_id: 1xoB5VYOd0XLjb-rQqqHWCkQvma4NytEd
-        path_local: /code/models/torchscript/predictor/fer/2/model.pt
+        path_local: /home/user/code/models/torchscript/predictor/fer/2/model.pt
       device:
         _target_: torch.device
         type: ${analyzer.device}
@@ -228,12 +228,87 @@ analyzer:
         - Neutral
         - Sadness
         - Surprise
+    au:
+      _target_: facetorch.analyzer.predictor.FacePredictor
+      downloader:
+        _target_: facetorch.downloader.DownloaderGDrive
+        file_id: 1uoVX9suSA5JVWTms3hEtJKzwO-CUR_jV
+        path_local: /home/user/opt/facetorch/models/torchscript/predictor/au/1/model.pt # str
+      device:
+        _target_: torch.device
+        type: ${analyzer.device} # str
+      preprocessor:
+        _target_: facetorch.analyzer.predictor.pre.PredictorPreProcessor
+        transform:
+          _target_: torchvision.transforms.Compose
+          transforms:
+            - _target_: torchvision.transforms.Resize
+              size: [224, 224] # List[int]
+            - _target_: torchvision.transforms.Normalize
+              mean: [0.485, 0.456, 0.406] # List[float]
+              std: [0.229, 0.224, 0.225] # List[float]
+        device: 
+          _target_: torch.device
+          type: ${analyzer.predictor.au.device.type}
+        optimize_transform: ${analyzer.optimize_transforms}
+        reverse_colors: False # bool
+      postprocessor:
+        _target_: facetorch.analyzer.predictor.post.PostMultiLabel
+        transform: None
+        device: 
+          _target_: torch.device
+          type: ${analyzer.predictor.au.device.type}
+        optimize_transform: ${analyzer.optimize_transforms}
+        dim: 1
+        threshold: 0.5
+        labels:
+        - inner_brow_raiser
+        - outer_brow_raiser
+        - brow_lowerer
+        - upper_lid_raiser
+        - cheek_raiser
+        - lid_tightener
+        - nose_wrinkler
+        - upper_lip_raiser
+        - nasolabial_deepener
+        - lip_corner_puller
+        - sharp_lip_puller
+        - dimpler
+        - lip_corner_depressor
+        - lower_lip_depressor
+        - chin_raiser
+        - lip_pucker
+        - tongue_show
+        - lip_stretcher
+        - lip_funneler
+        - lip_tightener
+        - lip_pressor
+        - lips_part
+        - jaw_drop
+        - mouth_stretch
+        - lip_bite
+        - nostril_dilator
+        - nostril_compressor
+        - left_inner_brow_raiser
+        - right_inner_brow_raiser
+        - left_outer_brow_raiser
+        - right_outer_brow_raiser
+        - left_brow_lowerer
+        - right_brow_lowerer
+        - left_cheek_raiser
+        - right_cheek_raiser
+        - left_upper_lip_raiser
+        - right_upper_lip_raiser
+        - left_nasolabial_deepener
+        - right_nasolabial_deepener
+        - left_dimpler
+        - right_dimpler
     deepfake:
       _target_: facetorch.analyzer.predictor.FacePredictor
       downloader:
         _target_: facetorch.downloader.DownloaderGDrive
         file_id: 1GjDTwQpvrkCjXOdiBy1oMkzm7nt-bXFg
-        path_local: /code/models/torchscript/predictor/deepfake/1/model.pt
+        path_local: /home/user/code/models/torchscript/predictor/deepfake/1/model.pt
       device:
         _target_: torch.device
         type: ${analyzer.device}
@@ -276,7 +351,7 @@ analyzer:
       downloader:
         _target_: facetorch.downloader.DownloaderGDrive
         file_id: 16gNFQdEH2nWvW3zTbdIAniKIbPAp6qBA
-        path_local: /code/models/torchscript/predictor/align/1/model.pt
+        path_local: /home/user/code/models/torchscript/predictor/align/1/model.pt
       device:
         _target_: torch.device
         type: ${analyzer.device}
@@ -314,7 +389,7 @@ analyzer:
       downloader_meta:
         _target_: facetorch.downloader.DownloaderGDrive
         file_id: 11tdAcFuSXqCCf58g52WT1Rpa8KuQwe2o
-        path_local: /code/data/3dmm/meta.pt
+        path_local: /home/user/code/data/3dmm/meta.pt
       image_size: 120
     draw_boxes:
       _target_: facetorch.analyzer.utilizer.draw.BoxDrawer
@@ -337,8 +412,8 @@ analyzer:
   logger:
     _target_: facetorch.logger.LoggerJsonFile
     name: facetorch
-    level: 20
-    path_file: /code/logs/facetorch/main.log
+    level: 10
+    path_file: /home/user/code/logs/facetorch/main.log
     json_format: '%(asctime)s %(levelname)s %(message)s'
 main:
   sleep: 3