f

models/SRFlow/srflow.py

@@ -22,7 +22,7 @@ def return_SRFlow_result(lr, divide, conf_path='models/SRFlow/code/confs/SRFlow_
     model, opt = load_model(conf_path)
 
     lr = PILToTensor()(lr).permute(1, 2, 0).numpy()
-    
+
     if not divide:
         lr *= 255
 
@@ -60,6 +60,9 @@ def return_SRFlow_result_from_tensor(lr_tensor, divide=True):
     for b in range(batch_size):
         lr_image = ToPILImage()(lr_tensor[b])
         sr_image = return_SRFlow_result(lr_image, divide)
+
+        if not divide:
+            sr_image /= 255
         sr_tensor = PILToTensor()(sr_image).unsqueeze(0)
         sr_list.append(sr_tensor)