Hugging Face's logo

Spaces:
PKUWilliamYang
/
StyleGANEX
Runtime error

App Files Community
1
StyleGANEX
File size: 1,209 Bytes 
01ad5b5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
 
dataset_paths = {
    'ffhq': 'data/train/ffhq/realign320x320/',
    'ffhq_test': 'data/train/ffhq/realign320x320test/',
    'ffhq1280': 'data/train/ffhq/realign1280x1280/',
    'ffhq1280_test': 'data/train/ffhq/realign1280x1280test/',    
    'ffhq_train_sketch': 'data/train/ffhq/realign640x640sketch/',
    'ffhq_test_sketch': 'data/train/ffhq/realign640x640sketchtest/',
    'ffhq_train_segmentation': 'data/train/ffhq/realign320x320mask/',
    'ffhq_test_segmentation': 'data/train/ffhq/realign320x320masktest/',
    'toonify_in': 'data/train/pixar/trainA/',
    'toonify_out': 'data/train/pixar/trainB/',
    'toonify_test_in': 'data/train/pixar/testA/',
    'toonify_test_out': 'data/train/testB/',
}

model_paths = {
    'stylegan_ffhq': 'pretrained_models/stylegan2-ffhq-config-f.pt',
    'ir_se50': 'pretrained_models/model_ir_se50.pth',
    'circular_face': 'pretrained_models/CurricularFace_Backbone.pth',
    'mtcnn_pnet': 'pretrained_models/mtcnn/pnet.npy',
    'mtcnn_rnet': 'pretrained_models/mtcnn/rnet.npy',
    'mtcnn_onet': 'pretrained_models/mtcnn/onet.npy',
    'shape_predictor': 'shape_predictor_68_face_landmarks.dat',
    'moco': 'pretrained_models/moco_v2_800ep_pretrain.pth.tar'
}