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

del try catch in unrelate codes

Browse Source
This commit is contained in:
Moeblack 2023-06-10 16:35:46 +08:00 committed by henryruhs
parent 692d097696
commit c3c01673ae
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
roop/enhancer.py
View File

@ -112,7 +112,7 @@ def normalize_face(face):
return face_in_tensor.unsqueeze(0).to(device)
def enhanced_face_in_tensor(face_in_tensor, codeformer_fidelity = 0.6):
def enhance_face_in_tensor(face_in_tensor, codeformer_fidelity = 0.6):
with torch.no_grad():
enhanced_face_in_tensor = get_code_former()(face_in_tensor, w=codeformer_fidelity, adain=True)[0]
return enhanced_face_in_tensor
@ -124,8 +124,8 @@ def convert_tensor_to_image(enhanced_face_in_tensor):
def restore_face(face_in_tensor):
enhanced_face_in_tensor = enhance_face_in_tensor(face_in_tensor)
try:
enhanced_face_in_tensor = enhanced_face_in_tensor(face_in_tensor)
restored_face = convert_tensor_to_image(enhanced_face_in_tensor)
del enhanced_face_in_tensor
except RuntimeError as error:
@ -137,13 +137,13 @@ def restore_face(face_in_tensor):
def process_frames(source_path: str, frame_paths: list[str], progress=None) -> None:
source_face = None
for frame_path in frame_paths:
frame = cv2.imread(frame_path)
try:
frame = cv2.imread(frame_path)
result = process_faces(source_face, frame)
cv2.imwrite(frame_path, result)
except Exception as exception:
print(exception)
pass
continue
if progress:
progress.update(1)