Add AdaFace conf file

aadityaverma · Nov 17, 2022 · 702822b · 702822b
1 parent ab12c7f
commit 702822b
Showing 1 changed file with 35 additions and 0 deletions.
diff --git a/conf/analyzer/predictor/verify/adaface_ir101_webface12m.yaml b/conf/analyzer/predictor/verify/adaface_ir101_webface12m.yaml
@@ -0,0 +1,35 @@
+_target_: facetorch.analyzer.predictor.FacePredictor
+
+downloader:
+  _target_: facetorch.downloader.DownloaderGDrive
+  file_id: 1WI-mP_0mGW31OHfriPUsuFS_usYh_W8p
+  path_local: /opt/facetorch/models/torchscript/predictor/verify/2/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: [112, 112] # 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.verify.device.type}
+  optimize_transform: ${analyzer.optimize_transforms}
+  reverse_colors: True # bool
+
+postprocessor:
+  _target_: facetorch.analyzer.predictor.post.PostEmbedder
+  transform: None
+  device: 
+    _target_: torch.device
+    type: ${analyzer.predictor.verify.device.type}
+  optimize_transform: ${analyzer.optimize_transforms}
+  labels: ["abstract"] # List