Update config.merged.gpu.yml

config.merged.gpu.yml

@@ -17,9 +17,11 @@ analyzer:
   detector:
     _target_: facetorch.analyzer.detector.FaceDetector
     downloader:
-      _target_: facetorch.downloader.DownloaderGDrive
+      _target_: facetorch.downloader.DownloaderHuggingFace
       file_id: 154x2VjmTQVqmowB0yZw4Uck7uQs2vVBs
-      path_local: /home/user/code/models/torchscript/detector/1/model.pt
+      path_local: /opt/facetorch/models/torchscript/detector/1/model.pt
+      repo_id: tomas-gajarsky/facetorch-detector-retinaface
+      filename: model.pt
     device:
       _target_: torch.device
       type: ${analyzer.device}
@@ -30,14 +32,8 @@ analyzer:
         _target_: torchvision.transforms.Compose
         transforms:
         - _target_: torchvision.transforms.Normalize
-          mean:
-          - 123.0
-          - 117.0
-          - 104.0
-          std:
-          - 1.0
-          - 1.0
-          - 1.0
+          mean: [123.0, 117.0, 104.0]
+          std:  [1.0, 1.0, 1.0]
       device:
         _target_: torch.device
         type: ${analyzer.device}
@@ -58,20 +54,12 @@ analyzer:
       prior_box:
         _target_: facetorch.analyzer.detector.post.PriorBox
         min_sizes:
-        - - 16
-          - 32
-        - - 64
-          - 128
-        - - 256
-          - 512
-        steps:
-        - 8
-        - 16
-        - 32
+        - [16, 32]
+        - [64, 128]
+        - [256, 512]
+        steps: [8, 16, 32]
         clip: false
-      variance:
-      - 0.1
-      - 0.2
+      variance: [0.1, 0.2]
       reverse_colors: ${analyzer.detector.reverse_colors}
       expand_box_ratio: 0.1
   unifier:
@@ -80,18 +68,10 @@ analyzer:
       _target_: torchvision.transforms.Compose
       transforms:
       - _target_: torchvision.transforms.Normalize
-        mean:
-        - -123.0
-        - -117.0
-        - -104.0
-        std:
-        - 255.0
-        - 255.0
-        - 255.0
+        mean: [-123.0, -117.0, -104.0]
+        std:  [255.0, 255.0, 255.0]
       - _target_: torchvision.transforms.Resize
-        size:
-        - 380
-        - 380
+        size: [380, 380]
     device:
       _target_: torch.device
       type: ${analyzer.device}
@@ -100,9 +80,11 @@ analyzer:
     embed:
       _target_: facetorch.analyzer.predictor.FacePredictor
       downloader:
-        _target_: facetorch.downloader.DownloaderGDrive
+        _target_: facetorch.downloader.DownloaderHuggingFace
         file_id: 19h3kqar1wlELAmM5hDyj9tlrUh8yjrCl
-        path_local: /home/user/code/models/torchscript/predictor/embed/1/model.pt
+        path_local: /opt/facetorch/models/torchscript/predictor/embed/1/model.pt
+        repo_id: tomas-gajarsky/facetorch-embed-resnet-50
+        filename: model.pt
       device:
         _target_: torch.device
         type: ${analyzer.device}
@@ -112,18 +94,10 @@ analyzer:
           _target_: torchvision.transforms.Compose
           transforms:
           - _target_: torchvision.transforms.Resize
-            size:
-            - 244
-            - 244
+            size: [244, 244]
           - _target_: torchvision.transforms.Normalize
-            mean:
-            - 0.485
-            - 0.456
-            - 0.406
-            std:
-            - 0.228
-            - 0.224
-            - 0.225
+            mean: [0.485, 0.456, 0.406]
+            std:  [0.228, 0.224, 0.225]
         device:
           _target_: torch.device
           type: ${analyzer.predictor.fer.device.type}
@@ -136,14 +110,15 @@ analyzer:
           _target_: torch.device
           type: ${analyzer.predictor.fer.device.type}
         optimize_transform: ${analyzer.optimize_transforms}
-        labels:
-        - abstract
+        labels: [abstract]
     verify:
       _target_: facetorch.analyzer.predictor.FacePredictor
       downloader:
-        _target_: facetorch.downloader.DownloaderGDrive
+        _target_: facetorch.downloader.DownloaderHuggingFace
         file_id: 1H-aPtFd9C5D7y1vzoWsObKAxeIBE9QPd
-        path_local: /home/user/code/models/torchscript/predictor/verify/1/model.pt
+        path_local: /opt/facetorch/models/torchscript/predictor/verify/1/model.pt
+        repo_id: tomas-gajarsky/facetorch-verify-adaface
+        filename: model.pt
       device:
         _target_: torch.device
         type: ${analyzer.device}
@@ -153,18 +128,10 @@ analyzer:
           _target_: torchvision.transforms.Compose
           transforms:
           - _target_: torchvision.transforms.Resize
-            size:
-            - 112
-            - 112
+            size: [112, 112]
           - _target_: torchvision.transforms.Normalize
-            mean:
-            - 0.485
-            - 0.456
-            - 0.406
-            std:
-            - 0.229
-            - 0.224
-            - 0.225
+            mean: [0.485, 0.456, 0.406]
+            std:  [0.229, 0.224, 0.225]
         device:
           _target_: torch.device
           type: ${analyzer.predictor.verify.device.type}
@@ -177,14 +144,15 @@ analyzer:
           _target_: torch.device
           type: ${analyzer.predictor.verify.device.type}
         optimize_transform: ${analyzer.optimize_transforms}
-        labels:
-        - abstract
+        labels: [abstract]
     fer:
       _target_: facetorch.analyzer.predictor.FacePredictor
       downloader:
-        _target_: facetorch.downloader.DownloaderGDrive
+        _target_: facetorch.downloader.DownloaderHuggingFace
         file_id: 1xoB5VYOd0XLjb-rQqqHWCkQvma4NytEd
-        path_local: /home/user/code/models/torchscript/predictor/fer/2/model.pt
+        path_local: /opt/facetorch/models/torchscript/predictor/fer/2/model.pt
+        repo_id: tomas-gajarsky/facetorch-fer-efficientnet-b2
+        filename: model.pt
       device:
         _target_: torch.device
         type: ${analyzer.device}
@@ -194,18 +162,10 @@ analyzer:
           _target_: torchvision.transforms.Compose
           transforms:
           - _target_: torchvision.transforms.Resize
-            size:
-            - 260
-            - 260
+            size: [260, 260]
           - _target_: torchvision.transforms.Normalize
-            mean:
-            - 0.485
-            - 0.456
-            - 0.406
-            std:
-            - 0.229
-            - 0.224
-            - 0.225
+            mean: [0.485, 0.456, 0.406]
+            std:  [0.229, 0.224, 0.225]
         device:
           _target_: torch.device
           type: ${analyzer.predictor.fer.device.type}
@@ -231,31 +191,33 @@ analyzer:
     au:
       _target_: facetorch.analyzer.predictor.FacePredictor
       downloader:
-        _target_: facetorch.downloader.DownloaderGDrive
+        _target_: facetorch.downloader.DownloaderHuggingFace
         file_id: 1uoVX9suSA5JVWTms3hEtJKzwO-CUR_jV
-        path_local: /home/user/opt/facetorch/models/torchscript/predictor/au/1/model.pt # str
+        path_local: /opt/facetorch/models/torchscript/predictor/au/1/model.pt
+        repo_id: tomas-gajarsky/facetorch-au-opengraph
+        filename: model.pt
       device:
         _target_: torch.device
-        type: ${analyzer.device} # str
+        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_: 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
+        reverse_colors: False           # bool
       postprocessor:
         _target_: facetorch.analyzer.predictor.post.PostMultiLabel
         transform: None
-        device: 
+        device:
           _target_: torch.device
           type: ${analyzer.predictor.au.device.type}
         optimize_transform: ${analyzer.optimize_transforms}
@@ -306,9 +268,11 @@ analyzer:
     va:
       _target_: facetorch.analyzer.predictor.FacePredictor
       downloader:
-        _target_: facetorch.downloader.DownloaderGDrive
+        _target_: facetorch.downloader.DownloaderHuggingFace
         file_id: 1Xl4ilNCU_DgKNhITrXb3UyQUUdm3VTKS
-        path_local: /home/user/code/models/torchscript/predictor/va/1/model.pt
+        path_local: /opt/facetorch/models/torchscript/predictor/va/1/model.pt
+        repo_id: tomas-gajarsky/facetorch-va-elim
+        filename: model.pt
       device:
         _target_: torch.device
         type: ${analyzer.device}
@@ -318,19 +282,11 @@ analyzer:
           _target_: torchvision.transforms.Compose
           transforms:
           - _target_: torchvision.transforms.Resize
-            size:
-            - 224
-            - 224
+            size: [224, 224]
             antialias: true
           - _target_: torchvision.transforms.Normalize
-            mean:
-            - 0.485
-            - 0.456
-            - 0.406
-            std:
-            - 0.229
-            - 0.224
-            - 0.225
+            mean: [0.485, 0.456, 0.406]
+            std:  [0.229, 0.224, 0.225]
         device:
           _target_: torch.device
           type: ${analyzer.predictor.va.device.type}
@@ -343,15 +299,15 @@ analyzer:
           _target_: torch.device
           type: ${analyzer.predictor.va.device.type}
         optimize_transform: ${analyzer.optimize_transforms}
-        labels:
-        - valence
-        - arousal
+        labels: [valence, arousal]
     deepfake:
       _target_: facetorch.analyzer.predictor.FacePredictor
       downloader:
-        _target_: facetorch.downloader.DownloaderGDrive
+        _target_: facetorch.downloader.DownloaderHuggingFace
         file_id: 1GjDTwQpvrkCjXOdiBy1oMkzm7nt-bXFg
-        path_local: /home/user/code/models/torchscript/predictor/deepfake/1/model.pt
+        path_local: /opt/facetorch/models/torchscript/predictor/deepfake/1/model.pt
+        repo_id: tomas-gajarsky/facetorch-deepfake-efficientnet-b7
+        filename: model.pt
       device:
         _target_: torch.device
         type: ${analyzer.device}
@@ -361,18 +317,10 @@ analyzer:
           _target_: torchvision.transforms.Compose
           transforms:
           - _target_: torchvision.transforms.Resize
-            size:
-            - 380
-            - 380
+            size: [380, 380]
           - _target_: torchvision.transforms.Normalize
-            mean:
-            - 0.485
-            - 0.456
-            - 0.406
-            std:
-            - 0.229
-            - 0.224
-            - 0.225
+            mean: [0.485, 0.456, 0.406]
+            std:  [0.229, 0.224, 0.225]
         device:
           _target_: torch.device
           type: ${analyzer.device}
@@ -385,16 +333,16 @@ analyzer:
           _target_: torch.device
           type: ${analyzer.device}
         optimize_transform: ${analyzer.optimize_transforms}
-        labels:
-        - Real
-        - Fake
+        labels: [Real, Fake]
         threshold: 0.7
     align:
       _target_: facetorch.analyzer.predictor.FacePredictor
       downloader:
-        _target_: facetorch.downloader.DownloaderGDrive
+        _target_: facetorch.downloader.DownloaderHuggingFace
         file_id: 16gNFQdEH2nWvW3zTbdIAniKIbPAp6qBA
-        path_local: /home/user/code/models/torchscript/predictor/align/1/model.pt
+        path_local: /opt/facetorch/models/torchscript/predictor/align/1/model.pt
+        repo_id: tomas-gajarsky/facetorch-align-synergynet
+        filename: model.pt
       device:
         _target_: torch.device
         type: ${analyzer.device}
@@ -404,9 +352,7 @@ analyzer:
           _target_: torchvision.transforms.Compose
           transforms:
           - _target_: torchvision.transforms.Resize
-            size:
-            - 120
-            - 120
+            size: [120, 120]
         device:
           _target_: torch.device
           type: ${analyzer.predictor.align.device.type}
@@ -419,8 +365,7 @@ analyzer:
           _target_: torch.device
           type: ${analyzer.predictor.align.device.type}
         optimize_transform: ${analyzer.optimize_transforms}
-        labels:
-        - abstract
+        labels: [abstract]
   utilizer:
     align:
       _target_: facetorch.analyzer.utilizer.align.Lmk3DMeshPose
@@ -430,9 +375,9 @@ analyzer:
         type: ${analyzer.device}
       optimize_transform: false
       downloader_meta:
-        _target_: facetorch.downloader.DownloaderGDrive
+        _target_: facetorch.downloader.DownloaderGDrive   # (unchanged)
         file_id: 11tdAcFuSXqCCf58g52WT1Rpa8KuQwe2o
-        path_local: /home/user/code/data/3dmm/meta.pt
+        path_local: /opt/facetorch/data/3dmm/meta.pt
       image_size: 120
     draw_boxes:
       _target_: facetorch.analyzer.utilizer.draw.BoxDrawer