From 702822b3060bfa72feee2581bb58543f151f6f9d Mon Sep 17 00:00:00 2001 From: tomas-gajarsky Date: Thu, 17 Nov 2022 23:20:43 +0100 Subject: [PATCH] Add AdaFace conf file --- .../verify/adaface_ir101_webface12m.yaml | 35 +++++++++++++++++++ 1 file changed, 35 insertions(+) create mode 100644 conf/analyzer/predictor/verify/adaface_ir101_webface12m.yaml diff --git a/conf/analyzer/predictor/verify/adaface_ir101_webface12m.yaml b/conf/analyzer/predictor/verify/adaface_ir101_webface12m.yaml new file mode 100644 index 0000000..c63487a --- /dev/null +++ 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