third_party/GPEN/__init_paths.py

'''
@paper: GAN Prior Embedded Network for Blind Face Restoration in the Wild (CVPR2021)
@author: yangxy ([email protected])
'''
import os.path as osp
import sys

def add_path(path):
    if path not in sys.path:
        sys.path.insert(0, path)

this_dir = osp.dirname(__file__)

path = osp.join(this_dir, 'face_detect')
add_path(path)

path = osp.join(this_dir, 'face_parse')
add_path(path)

path = osp.join(this_dir, 'face_model')
add_path(path)

path = osp.join(this_dir, 'sr_model')
add_path(path)

path = osp.join(this_dir, 'training')
add_path(path)

path = osp.join(this_dir, 'training/loss')
add_path(path)

path = osp.join(this_dir, 'training/data_loader')
add_path(path)