change heat to 0.7

models/SRFlow/example.py

@@ -1,14 +0,0 @@
-import matplotlib.pyplot as plt
-import sys
-
-sys.path.append('SRFlow')
-from srflow import *
-
-# Example usage of the function
-path = 'images/demo.png'
-output = return_SRFlow_result(path) 
-print(output.shape)
-
-plt.imshow(output)
-# plt.axis('off')  
-plt.show()

models/SRFlow/srflow.py

@@ -6,7 +6,7 @@ from SRFlow.code import imread, impad, load_model, t, rgb
 from PIL import Image
 from torchvision.transforms import PILToTensor
 
-def return_SRFlow_result(lr, conf_path='models/SRFlow/code/confs/SRFlow_DF2K_4X.yml', heat=0.6):
+def return_SRFlow_result(lr, conf_path='models/SRFlow/code/confs/SRFlow_DF2K_4X.yml', heat=0.7):
     """
     Apply Super-Resolution using SRFlow model to the input LR (low-resolution) image.