feat 2倍超分

This commit is contained in:
zhouchengrong
2024-04-11 11:16:40 +08:00
parent 847c4adecf
commit e6f9a066ad

View File

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