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 = [