Spaces:
Sleeping
Sleeping
File size: 1,098 Bytes
95b1715 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
from dataclasses import dataclass, asdict, fields
models_dir = "pretrained_models/"
@dataclass
class DefaultPathsClass:
psp_path: str = models_dir + "psp_ffhq_encode.pt"
e4e_path: str = models_dir + "e4e_ffhq_encode.pt"
farl_path:str = models_dir + "face_parsing.farl.lapa.main_ema_136500_jit191.pt"
mobile_net_pth: str = models_dir + "mobilenet0.25_Final.pth"
ir_se50_path: str = models_dir + "model_ir_se50.pth"
stylegan_weights: str = models_dir + "stylegan2-ffhq-config-f.pt"
stylegan_car_weights: str = models_dir + "stylegan2-car-config-f-new.pkl"
stylegan_weights_pkl: str = models_dir + "stylegan2-ffhq-config-f.pkl"
arcface_model_path: str = models_dir + "iresnet50-7f187506.pth"
moco: str = models_dir + "moco_v2_800ep_pretrain.pt"
curricular_face_path: str = models_dir + "CurricularFace_Backbone.pth"
mtcnn: str = models_dir + "mtcnn"
landmark: str = models_dir + "79999_iter.pth"
def __iter__(self):
for field in fields(self):
yield field.name, getattr(self, field.name)
DefaultPaths = DefaultPathsClass()
|