k

models/SRFlow/code/options/options.py

@@ -79,7 +79,7 @@ def parse(opt_path, is_train=True):
             opt['train']['lr_steps'] = [int(x * niter) for x in opt['train']['lr_steps_rel']]
         if 'lr_steps_inverse_rel' in opt['train']:
             opt['train']['lr_steps_inverse'] = [int(x * niter) for x in opt['train']['lr_steps_inverse_rel']]
-        print(opt['train'])
+        # print(opt['train'])
 
     return opt
 

models/SRFlow/srflow.py

@@ -68,7 +68,7 @@ def return_SRFlow_result_from_tensor(lr_tensor, divide=True):
     sr_tensor = torch.cat(sr_list, dim=0)
 
     if not divide:
-        sr_tensor /= 255
+        sr_tensor /= 255.0
         
     return sr_tensor