facexlib

1
import glob
2
import os
3

4
import facexlib.utils.face_restoration_helper as face_restoration_helper
5

6

7
def crop_one_img(img, save_cropped_path=None):
8
    FaceRestoreHelper.clean_all()
9
    FaceRestoreHelper.read_image(img)
10
    # get face landmarks
11
    FaceRestoreHelper.get_face_landmarks_5()
12
    FaceRestoreHelper.align_warp_face(save_cropped_path)
13

14

15
if __name__ == '__main__':
16
    # initialize face helper
17
    FaceRestoreHelper = face_restoration_helper.FaceRestoreHelper(upscale_factor=1)
18

19
    img_paths = glob.glob('/home/wxt/Projects/test/*')
20
    save_path = 'test'
21
    for idx, path in enumerate(img_paths):
22
        print(idx, path)
23
        file_name = os.path.basename(path)
24
        save_cropped_path = os.path.join(save_path, file_name)
25
        crop_one_img(path, save_cropped_path=save_cropped_path)
26