Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

add facehires

Browse Source
This commit is contained in:
Vladimir Mandic
2024-03-10 13:17:08 -04:00
parent 0552b8ce72
commit 0f7bb8a1f4
12 changed files with 201 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/postprocess/codeformer_model.py
+2 -2
View File
@@ -66,7 +66,7 @@ def setup_model(dirname):
self.face_helper.face_det.to(device) # pylint: disable=no-member
self.face_helper.face_parse.to(device)
def restore(self, np_image, w=None):
def restore(self, np_image, p=None, w=None): # pylint: disable=unused-argument
from torchvision.transforms.functional import normalize
from basicsr.utils import img2tensor, tensor2img
np_image = np_image[:, :, ::-1]
@@ -90,7 +90,7 @@ def setup_model(dirname):
del output
devices.torch_gc()
except Exception as e:
shared.log.error(f'CodeForomer error: {e}')
shared.log.error(f'CodeFormer error: {e}')
restored_face = tensor2img(cropped_face_t, rgb2bgr=True, min_max=(-1, 1))
restored_face = restored_face.astype('uint8')
self.face_helper.add_restored_face(restored_face)