Spaces:
Running
Running
tomas-gajarsky
commited on
Commit
•
58754c3
1
Parent(s):
08e6707
Add Action Unit model to config
Browse files- config.merged.yml +75 -0
config.merged.yml
CHANGED
@@ -228,6 +228,81 @@ analyzer:
|
|
228 |
- Neutral
|
229 |
- Sadness
|
230 |
- Surprise
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
231 |
deepfake:
|
232 |
_target_: facetorch.analyzer.predictor.FacePredictor
|
233 |
downloader:
|
|
|
228 |
- Neutral
|
229 |
- Sadness
|
230 |
- Surprise
|
231 |
+
au:
|
232 |
+
_target_: facetorch.analyzer.predictor.FacePredictor
|
233 |
+
downloader:
|
234 |
+
_target_: facetorch.downloader.DownloaderGDrive
|
235 |
+
file_id: 1uoVX9suSA5JVWTms3hEtJKzwO-CUR_jV
|
236 |
+
path_local: /opt/facetorch/models/torchscript/predictor/au/1/model.pt # str
|
237 |
+
device:
|
238 |
+
_target_: torch.device
|
239 |
+
type: ${analyzer.device} # str
|
240 |
+
preprocessor:
|
241 |
+
_target_: facetorch.analyzer.predictor.pre.PredictorPreProcessor
|
242 |
+
transform:
|
243 |
+
_target_: torchvision.transforms.Compose
|
244 |
+
transforms:
|
245 |
+
- _target_: torchvision.transforms.Resize
|
246 |
+
size: [224, 224] # List[int]
|
247 |
+
- _target_: torchvision.transforms.Normalize
|
248 |
+
mean: [0.485, 0.456, 0.406] # List[float]
|
249 |
+
std: [0.229, 0.224, 0.225] # List[float]
|
250 |
+
device:
|
251 |
+
_target_: torch.device
|
252 |
+
type: ${analyzer.predictor.au.device.type}
|
253 |
+
optimize_transform: ${analyzer.optimize_transforms}
|
254 |
+
reverse_colors: False # bool
|
255 |
+
postprocessor:
|
256 |
+
_target_: facetorch.analyzer.predictor.post.PostMultiLabel
|
257 |
+
transform: None
|
258 |
+
device:
|
259 |
+
_target_: torch.device
|
260 |
+
type: ${analyzer.predictor.au.device.type}
|
261 |
+
optimize_transform: ${analyzer.optimize_transforms}
|
262 |
+
dim: 1
|
263 |
+
threshold: 0.5
|
264 |
+
labels:
|
265 |
+
- inner_brow_raiser
|
266 |
+
- outer_brow_raiser
|
267 |
+
- brow_lowerer
|
268 |
+
- upper_lid_raiser
|
269 |
+
- cheek_raiser
|
270 |
+
- lid_tightener
|
271 |
+
- nose_wrinkler
|
272 |
+
- upper_lip_raiser
|
273 |
+
- nasolabial_deepener
|
274 |
+
- lip_corner_puller
|
275 |
+
- sharp_lip_puller
|
276 |
+
- dimpler
|
277 |
+
- lip_corner_depressor
|
278 |
+
- lower_lip_depressor
|
279 |
+
- chin_raiser
|
280 |
+
- lip_pucker
|
281 |
+
- tongue_show
|
282 |
+
- lip_stretcher
|
283 |
+
- lip_funneler
|
284 |
+
- lip_tightener
|
285 |
+
- lip_pressor
|
286 |
+
- lips_part
|
287 |
+
- jaw_drop
|
288 |
+
- mouth_stretch
|
289 |
+
- lip_bite
|
290 |
+
- nostril_dilator
|
291 |
+
- nostril_compressor
|
292 |
+
- left_inner_brow_raiser
|
293 |
+
- right_inner_brow_raiser
|
294 |
+
- left_outer_brow_raiser
|
295 |
+
- right_outer_brow_raiser
|
296 |
+
- left_brow_lowerer
|
297 |
+
- right_brow_lowerer
|
298 |
+
- left_cheek_raiser
|
299 |
+
- right_cheek_raiser
|
300 |
+
- left_upper_lip_raiser
|
301 |
+
- right_upper_lip_raiser
|
302 |
+
- left_nasolabial_deepener
|
303 |
+
- right_nasolabial_deepener
|
304 |
+
- left_dimpler
|
305 |
+
- right_dimpler
|
306 |
deepfake:
|
307 |
_target_: facetorch.analyzer.predictor.FacePredictor
|
308 |
downloader:
|