fi0r1/roop
1
0
Fork 0
mirror of https://github.com/s0md3v/roop.git synced 2025-12-16 03:27:19 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

methods rename

Browse Source
This commit is contained in:
Moeblack 2023-06-10 16:21:50 +08:00 committed by henryruhs
parent abb8faa159
commit bee5829f2d
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
roop/enhancer.py
View File

@ -112,25 +112,25 @@ def normalize_face(face):
return face_in_tensor.unsqueeze(0).to(device) return face_in_tensor.unsqueeze(0).to(device)
def enhance_face_in_tensor(face_in_tensor, codeformer_fidelity = 0.6): def enhanced_face_in_tensor(face_in_tensor, codeformer_fidelity = 0.6):
with torch.no_grad(): with torch.no_grad():
output = get_code_former()(face_in_tensor, w=codeformer_fidelity, adain=True)[0] enhanced_face_in_tensor = get_code_former()(face_in_tensor, w=codeformer_fidelity, adain=True)[0]
return output return enhanced_face_in_tensor
def postprocess_output(output): def convert_tensor_to_image(enhanced_face_in_tensor):
restored_face = tensor2img(output, rgb2bgr=True, min_max=(-1, 1)) restored_face = tensor2img(enhanced_face_in_tensor, rgb2bgr=True, min_max=(-1, 1))
return restored_face.astype("uint8") return restored_face.astype("uint8")
def restore_face(face_in_tensor): def restore_face(face_in_tensor):
try: try:
output = enhance_face_in_tensor(face_in_tensor) enhanced_face_in_tensor = enhanced_face_in_tensor(face_in_tensor)
restored_face = postprocess_output(output) restored_face = convert_tensor_to_image(enhanced_face_in_tensor)
del output del enhanced_face_in_tensor
except RuntimeError as error: except RuntimeError as error:
print(f"Failed inference for CodeFormer-tensor: {error}") print(f"Failed inference for CodeFormer-tensor: {error}")
restored_face = postprocess_output(face_in_tensor) restored_face = convert_tensor_to_image(face_in_tensor)
return restored_face return restored_face