l

models/SRFlow/srflow.py

@@ -24,7 +24,7 @@ def return_SRFlow_result(lr, divide, conf_path='models/SRFlow/code/confs/SRFlow_
     lr = PILToTensor()(lr).permute(1, 2, 0).numpy()
 
     if not divide:
-        lr *= 255
+        lr = int(round(255*lr))
 
     scale = opt['scale']
     pad_factor = 2
@@ -68,9 +68,9 @@ def return_SRFlow_result_from_tensor(lr_tensor, divide=True):
     sr_tensor = torch.cat(sr_list, dim=0)
 
     if not divide:
-        sr_tensor /= 255.0
-        
-    return sr_tensor
+        sr_tensor /= 255
+  
+    return sr_tensor 
 
 if __name__ == '__main__':
     lr = Image.open('images/demo.png')