| """This script is to generate training list files for Deep3DFaceRecon_pytorch
|
| """
|
|
|
| import os
|
|
|
|
|
| def write_list(lms_list, imgs_list, msks_list, mode='train',save_folder='datalist', save_name=''):
|
| save_path = os.path.join(save_folder, mode)
|
| if not os.path.isdir(save_path):
|
| os.makedirs(save_path)
|
| with open(os.path.join(save_path, save_name + 'landmarks.txt'), 'w') as fd:
|
| fd.writelines([i + '\n' for i in lms_list])
|
|
|
| with open(os.path.join(save_path, save_name + 'images.txt'), 'w') as fd:
|
| fd.writelines([i + '\n' for i in imgs_list])
|
|
|
| with open(os.path.join(save_path, save_name + 'masks.txt'), 'w') as fd:
|
| fd.writelines([i + '\n' for i in msks_list])
|
|
|
|
|
| def check_list(rlms_list, rimgs_list, rmsks_list):
|
| lms_list, imgs_list, msks_list = [], [], []
|
| for i in range(len(rlms_list)):
|
| flag = 'false'
|
| lm_path = rlms_list[i]
|
| im_path = rimgs_list[i]
|
| msk_path = rmsks_list[i]
|
| if os.path.isfile(lm_path) and os.path.isfile(im_path) and os.path.isfile(msk_path):
|
| flag = 'true'
|
| lms_list.append(rlms_list[i])
|
| imgs_list.append(rimgs_list[i])
|
| msks_list.append(rmsks_list[i])
|
| print(i, rlms_list[i], flag)
|
| return lms_list, imgs_list, msks_list
|
|
|
