From e6f9a066ad16a90c6de0763591acd51faac59243 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Thu, 11 Apr 2024 11:16:40 +0800 Subject: [PATCH] =?UTF-8?q?feat=202=E5=80=8D=E8=B6=85=E5=88=86?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/super_resolution/service.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/app/service/super_resolution/service.py b/app/service/super_resolution/service.py index 82210c3..1cfd386 100644 --- a/app/service/super_resolution/service.py +++ b/app/service/super_resolution/service.py @@ -67,6 +67,10 @@ class SuperResolution: @RunTime def sr_result(self): sample = self.read_image() + if self.sr_xn == 2: + new_shape = (sample.shape[0] // self.sr_xn, sample.shape[1] // self.sr_xn) + sample = cv2.resize(sample, new_shape) + print(new_shape) sample = np.transpose(sample if sample.shape[2] == 1 else sample[:, :, [2, 1, 0]], (2, 0, 1)) sample = torch.from_numpy(sample).float().unsqueeze(0).numpy() inputs = [