Spaces:
Runtime error
Runtime error
| dataset_paths = { | |
| # Face Datasets (In the paper: FFHQ - train, CelebAHQ - test) | |
| 'ffhq': '', | |
| 'celeba_test': '', | |
| # Cars Dataset (In the paper: Stanford cars) | |
| 'cars_train': '', | |
| 'cars_test': '', | |
| # Horse Dataset (In the paper: LSUN Horse) | |
| 'horse_train': '', | |
| 'horse_test': '', | |
| # Church Dataset (In the paper: LSUN Church) | |
| 'church_train': '', | |
| 'church_test': '', | |
| # Cats Dataset (In the paper: LSUN Cat) | |
| 'cats_train': '', | |
| 'cats_test': '' | |
| } | |
| model_paths = { | |
| 'stylegan_ffhq': 'pretrained_models/stylegan2-ffhq-config-f.pt', | |
| 'ir_se50': 'pretrained_models/model_ir_se50.pth', | |
| 'shape_predictor': 'pretrained_models/shape_predictor_68_face_landmarks.dat', | |
| 'moco': 'pretrained_models/moco_v2_800ep_pretrain.pth' | |
| } | |