search

sxl142 / TEx-Face

34 stars 2 forks source link

Missing some files~ #1

Open JackAILab opened 4 months ago

JackAILab commented 4 months ago

AssertionError: ./HFGI3D-main/test_celeba_img_dataset is not a valid directory

Could you provide the datasets or the link? {

Face Datasets (In the paper: FFHQ - train, CelebAHQ - test)

'ffhq': './ffhq/final_crops',
'celeba_test': './CelebAMask-HQ/HQ_test',

}

Besides, the pre-trained models seems to be missed~ 'shape_predictor': 'pretrained_models/shape_predictor_68_face_landmarks.dat', 'moco': 'pretrained_models/moco_v2_800ep_pretrain.pth'

Thank you for your nice work!!!

sxl142 commented 4 months ago

'test_celeba_img_dataset', https://drive.google.com/file/d/1bF1fIUOZjK4JRDQKn1qhvIMU_MWxpwa_/view?usp=sharing 'shape_predictor' and 'moco': i do not use these .pth files. You can download these from https://github.com/omertov/encoder4editing