GPEN/__init_paths.py at main · vkvickkey/GPEN

Repository navigation

'''

@paper: GAN Prior Embedded Network for Blind Face Restoration in the Wild (CVPR2021)

@author: yangxy (yangtao9009@gmail.com)

'''

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)