Commit 4d2c0c7a authored by AUTOMATIC's avatar AUTOMATIC

undo CodeFormer's upscaling of images with dimensions less than 512.

parent decbbe81
import os import os
import sys import sys
import traceback import traceback
import cv2
import torch import torch
from modules import shared from modules import shared
...@@ -68,6 +70,8 @@ def setup_codeformer(): ...@@ -68,6 +70,8 @@ def setup_codeformer():
def restore(self, np_image, w=None): def restore(self, np_image, w=None):
np_image = np_image[:, :, ::-1] np_image = np_image[:, :, ::-1]
original_resolution = np_image.shape[0:2]
net, face_helper = self.create_models() net, face_helper = self.create_models()
face_helper.clean_all() face_helper.clean_all()
face_helper.read_image(np_image) face_helper.read_image(np_image)
...@@ -96,6 +100,10 @@ def setup_codeformer(): ...@@ -96,6 +100,10 @@ def setup_codeformer():
restored_img = face_helper.paste_faces_to_input_image() restored_img = face_helper.paste_faces_to_input_image()
restored_img = restored_img[:, :, ::-1] restored_img = restored_img[:, :, ::-1]
if original_resolution != restored_img.shape[0:2]:
restored_img = cv2.resize(restored_img, (0, 0), fx=original_resolution[1]/restored_img.shape[1], fy=original_resolution[0]/restored_img.shape[0], interpolation=cv2.INTER_LINEAR)
return restored_img return restored_img
global have_codeformer global have_codeformer
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment