feat 2倍超分
This commit is contained in:
@@ -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 = [
|
||||
|
||||
Reference in New Issue
Block a user