From 4d2c0c7a72dcb4bb5714cd4f784a7058f7dee6ed Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Sat, 10 Sep 2022 13:53:10 +0300
Subject: undo CodeFormer's upscaling of images with dimensions less than 512.

---
 modules/codeformer_model.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/modules/codeformer_model.py b/modules/codeformer_model.py
index 82b45b15..946b4a30 100644
--- a/modules/codeformer_model.py
+++ b/modules/codeformer_model.py
@@ -1,6 +1,8 @@
 import os
 import sys
 import traceback
+
+import cv2
 import torch
 
 from modules import shared
@@ -68,6 +70,8 @@ def setup_codeformer():
             def restore(self, np_image, w=None):
                 np_image = np_image[:, :, ::-1]
 
+                original_resolution = np_image.shape[0:2]
+
                 net, face_helper = self.create_models()
                 face_helper.clean_all()
                 face_helper.read_image(np_image)
@@ -96,6 +100,10 @@ def setup_codeformer():
 
                 restored_img = face_helper.paste_faces_to_input_image()
                 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
 
         global have_codeformer
-- 
cgit v1.2.1