zhiweili commited on
Commit
f590c37
·
1 Parent(s): 55c1cea

fix output size

Browse files
Files changed (1) hide show
  1. app_enhance.py +1 -1
app_enhance.py CHANGED
@@ -70,7 +70,7 @@ def enhance_image(
70
  # h, w = img.shape[0:2]
71
  # output = cv2.resize(output, (int(w * scale / 2), int(h * scale / 2)), interpolation=interpolation)
72
 
73
- h, w = img.shape[0:2]
74
  max_size = 3480
75
  if h > max_size:
76
  w = int(w * max_size / h)
 
70
  # h, w = img.shape[0:2]
71
  # output = cv2.resize(output, (int(w * scale / 2), int(h * scale / 2)), interpolation=interpolation)
72
 
73
+ h, w = output.shape[0:2]
74
  max_size = 3480
75
  if h > max_size:
76
  w = int(w * max_size / h)