From fee48adff3d66960cb17836f3f4efbfd0c8740c4 Mon Sep 17 00:00:00 2001 From: melMass Date: Wed, 5 Jul 2023 21:09:00 +0200 Subject: [PATCH] =?UTF-8?q?fix:=20=F0=9F=90=9B=20bg=20upscaler=20in=20gfpg?= =?UTF-8?q?an?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit I was just not returning from tensor2np 🦀☠️ --- requirements.txt | 3 ++- utils.py | 9 +-------- 2 files changed, 3 insertions(+), 9 deletions(-) diff --git a/requirements.txt b/requirements.txt index fca2f7a..4e49579 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,4 +6,5 @@ ifnude==0.0.3 insightface==0.7.3 mmcv==2.0.0 mmdet==3.0.0 -rembg==2.0.37 \ No newline at end of file +rembg==2.0.37 +facexlib==0.3.0 \ No newline at end of file diff --git a/utils.py b/utils.py index 1144ff9..50fb5d7 100644 --- a/utils.py +++ b/utils.py @@ -73,14 +73,7 @@ def np2tensor(img_np): def tensor2np(tensor: torch.Tensor) -> np.ndarray: - np.clip(255.0 * tensor.cpu().numpy().squeeze(), 0, 255).astype(np.uint8) + return np.clip(255.0 * tensor.cpu().numpy().squeeze(), 0, 255).astype(np.uint8) - -def img_tensor_to_np(img_tensor): - img_tensor = img_tensor.clone() - img_tensor = img_tensor * 255.0 - return img_tensor.squeeze(0).numpy().astype(np.float32) - -