9 lines
388 B
Python
9 lines
388 B
Python
model_paths = {
|
|
'ir_se50': 'pretrained_models/model_ir_se50.pth',
|
|
'circular_face': 'pretrained_models/CurricularFace_Backbone.pth',
|
|
'shape_predictor': 'shape_predictor_68_face_landmarks.dat',
|
|
'moco': 'pretrained_models/moco_v2_800ep_pretrain.pth.tar',
|
|
'invert_predictor_faces': 'pretrained_models/invertibility_faces_sg2.pt',
|
|
'segmenter_faces': 'pretrained_models/79999_iter.pth'
|
|
}
|