File size: 509 Bytes
9ce90b2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
from dataclasses import dataclass, field
from typing import Dict, Union

PARAM = Dict[str, Union[int,float]]
WEIGHT = str
NETWORK = Dict[str, Union[PARAM, WEIGHT]]

@dataclass
class TaskConfig:
    localize            : Dict[str, NETWORK]
    check_liveness      : Dict[str, NETWORK]
    check_mask          : Dict[str, NETWORK]
    estimate_headpose   : Dict[str, NETWORK]
    extract_vector      : Dict[str, NETWORK]
    extract_emotion     : Dict[str, NETWORK]
    extract_agegender   : Dict[str, NETWORK]