File size: 256 Bytes
03da825 |
1 2 3 4 5 6 7 8 9 10 11 12 |
from utils.registry import MODEL_REGISTRY
# Register additional modules
import additional_modules.gfpgan.gfpganv1_clean_arch
def get_model(opt):
MODEL_REGISTRY.scan_and_register()
return MODEL_REGISTRY.get(opt['model_class'])(**opt['params'])
|