NSFW_Check / dataclass.py
tanlocc's picture
Upload dataclass.py
9ce90b2
raw
history blame contribute delete
509 Bytes
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]