Add valence arousal model to config

config.merged.yml

@@ -303,6 +303,49 @@ analyzer:
         - right_nasolabial_deepener
         - left_dimpler
         - right_dimpler
+    va:
+      _target_: facetorch.analyzer.predictor.FacePredictor
+      downloader:
+        _target_: facetorch.downloader.DownloaderGDrive
+        file_id: 1Xl4ilNCU_DgKNhITrXb3UyQUUdm3VTKS
+        path_local: /home/user/code/models/torchscript/predictor/va/1/model.pt
+      device:
+        _target_: torch.device
+        type: ${analyzer.device}
+      preprocessor:
+        _target_: facetorch.analyzer.predictor.pre.PredictorPreProcessor
+        transform:
+          _target_: torchvision.transforms.Compose
+          transforms:
+          - _target_: torchvision.transforms.Resize
+            size:
+            - 224
+            - 224
+            antialias: true
+          - _target_: torchvision.transforms.Normalize
+            mean:
+            - 0.485
+            - 0.456
+            - 0.406
+            std:
+            - 0.229
+            - 0.224
+            - 0.225
+        device:
+          _target_: torch.device
+          type: ${analyzer.predictor.va.device.type}
+        optimize_transform: ${analyzer.optimize_transforms}
+        reverse_colors: false
+      postprocessor:
+        _target_: facetorch.analyzer.predictor.post.PostLabelConfidencePairs
+        transform: None
+        device:
+          _target_: torch.device
+          type: ${analyzer.predictor.va.device.type}
+        optimize_transform: ${analyzer.optimize_transforms}
+        labels:
+        - valence
+        - arousal
     deepfake:
       _target_: facetorch.analyzer.predictor.FacePredictor
       downloader: