kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

update

Browse Source
This commit is contained in:
Qing 2023-03-31 22:17:38 +08:00
parent 6579977439
commit c3cbf8d4a2
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
lama_cleaner/plugins/gfpgan_plugin.py
View File

@ -17,9 +17,11 @@ class GFPGANPlugin(BasePlugin):
model_path = download_model(url, model_md5) model_path = download_model(url, model_md5)
logger.info(f"GFPGAN model path: {model_path}") logger.info(f"GFPGAN model path: {model_path}")
face_det_device = "cpu" if "cuda" in str(device) else device
import facexlib import facexlib
if hasattr(facexlib.detection.retinaface, "device"): if hasattr(facexlib.detection.retinaface, "device"):
facexlib.detection.retinaface.device = "cpu" facexlib.detection.retinaface.device = device
# Use GFPGAN for face enhancement # Use GFPGAN for face enhancement
self.face_enhancer = MyGFPGANer( self.face_enhancer = MyGFPGANer(
@ -30,9 +32,10 @@ class GFPGANPlugin(BasePlugin):
device=device, device=device,
bg_upsampler=upscaler.model if upscaler is not None else None, bg_upsampler=upscaler.model if upscaler is not None else None,
) )
self.face_enhancer.face_helper.face_det.mean_tensor.to("cpu") self.face_enhancer.face_helper.face_det.mean_tensor.to(device)
self.face_enhancer.face_helper.face_det = self.face_enhancer.face_helper.face_det.to("cpu") self.face_enhancer.face_helper.face_det = (
self.face_enhancer.face_helper.face_det.to(device)
)
def __call__(self, rgb_np_img, files, form): def __call__(self, rgb_np_img, files, form):
weight = 0.5 weight = 0.5