m

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 = int(round(255*lr))
+        lr = np.round(255 * lr).astype(np.uint8) 
 
     scale = opt['scale']
     pad_factor = 2