Spaces:
Running
Running
tomas-gajarsky
commited on
Commit
·
2d095b3
1
Parent(s):
c7bdfa0
Add valence arousal model config
Browse files- config.merged.gpu.yml +43 -0
config.merged.gpu.yml
CHANGED
@@ -303,6 +303,49 @@ analyzer:
|
|
303 |
- right_nasolabial_deepener
|
304 |
- left_dimpler
|
305 |
- right_dimpler
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
306 |
deepfake:
|
307 |
_target_: facetorch.analyzer.predictor.FacePredictor
|
308 |
downloader:
|
|
|
303 |
- right_nasolabial_deepener
|
304 |
- left_dimpler
|
305 |
- right_dimpler
|
306 |
+
va:
|
307 |
+
_target_: facetorch.analyzer.predictor.FacePredictor
|
308 |
+
downloader:
|
309 |
+
_target_: facetorch.downloader.DownloaderGDrive
|
310 |
+
file_id: 1Xl4ilNCU_DgKNhITrXb3UyQUUdm3VTKS
|
311 |
+
path_local: /home/user/code/models/torchscript/predictor/va/1/model.pt
|
312 |
+
device:
|
313 |
+
_target_: torch.device
|
314 |
+
type: ${analyzer.device}
|
315 |
+
preprocessor:
|
316 |
+
_target_: facetorch.analyzer.predictor.pre.PredictorPreProcessor
|
317 |
+
transform:
|
318 |
+
_target_: torchvision.transforms.Compose
|
319 |
+
transforms:
|
320 |
+
- _target_: torchvision.transforms.Resize
|
321 |
+
size:
|
322 |
+
- 224
|
323 |
+
- 224
|
324 |
+
antialias: true
|
325 |
+
- _target_: torchvision.transforms.Normalize
|
326 |
+
mean:
|
327 |
+
- 0.485
|
328 |
+
- 0.456
|
329 |
+
- 0.406
|
330 |
+
std:
|
331 |
+
- 0.229
|
332 |
+
- 0.224
|
333 |
+
- 0.225
|
334 |
+
device:
|
335 |
+
_target_: torch.device
|
336 |
+
type: ${analyzer.predictor.va.device.type}
|
337 |
+
optimize_transform: ${analyzer.optimize_transforms}
|
338 |
+
reverse_colors: false
|
339 |
+
postprocessor:
|
340 |
+
_target_: facetorch.analyzer.predictor.post.PostLabelConfidencePairs
|
341 |
+
transform: None
|
342 |
+
device:
|
343 |
+
_target_: torch.device
|
344 |
+
type: ${analyzer.predictor.va.device.type}
|
345 |
+
optimize_transform: ${analyzer.optimize_transforms}
|
346 |
+
labels:
|
347 |
+
- valence
|
348 |
+
- arousal
|
349 |
deepfake:
|
350 |
_target_: facetorch.analyzer.predictor.FacePredictor
|
351 |
downloader:
|