fi0r1/roop
1
0
Fork 0
mirror of https://github.com/s0md3v/roop.git synced 2025-12-06 18:08:29 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Minor fixes

Browse Source
This commit is contained in:
henryruhs 2023-06-13 18:57:19 +02:00
parent 624cf45978
commit 1670ec6378
2 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
roop/processors/frame/core.py
View File

@ -2,7 +2,6 @@ import sys
import importlib
import threading
from typing import Any, List
from tqdm import tqdm
import roop
@ -10,7 +9,7 @@ import roop
FRAME_PROCESSORS_MODULES = None
def load_processor_module(frame_processor: str)-> Any:
def load_frame_processor_module(frame_processor: str)-> Any:
try:
frame_processor_module = importlib.import_module(f'roop.processors.frame.{frame_processor}')
except ImportError:
@ -23,7 +22,7 @@ def get_frame_processors_modules(frame_processors):
if FRAME_PROCESSORS_MODULES is None:
FRAME_PROCESSORS_MODULES = []
for frame_processor in frame_processors:
frame_processor_module = load_processor_module(frame_processor)
frame_processor_module = load_frame_processor_module(frame_processor)
FRAME_PROCESSORS_MODULES.append(frame_processor_module)
return FRAME_PROCESSORS_MODULES

2
roop/processors/frame/face_enhancer.py
View File

@ -1,9 +1,7 @@
from typing import List
import cv2
import torch
import threading
from torchvision.transforms.functional import normalize
from codeformer.facelib.utils.face_restoration_helper import FaceRestoreHelper
from codeformer.basicsr.utils.registry import ARCH_REGISTRY