Fediory commited on
Commit
068f0e2
·
1 Parent(s): 2b1d6b5

fix: brisque

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -36,11 +36,11 @@ def process_image(input_img,score,model_path,gamma=1.0,alpha_s=1.0,alpha_i=1.0):
36
  output = output[:, :, :h, :w]
37
  enhanced_img = transforms.ToPILImage()(output.squeeze(0))
38
  if score == 'Yes':
39
- im1 = enhanced_img
40
- score_brisque = brisque.score(im1)
41
  im1 = np.array(im1)
42
  score_niqe = calculate_niqe(im1)
43
- return enhanced_img,score_niqe,score_brisque
44
  else:
45
  return enhanced_img,0,0
46
 
 
36
  output = output[:, :, :h, :w]
37
  enhanced_img = transforms.ToPILImage()(output.squeeze(0))
38
  if score == 'Yes':
39
+ # im1 = enhanced_img
40
+ # score_brisque = brisque.score(im1)
41
  im1 = np.array(im1)
42
  score_niqe = calculate_niqe(im1)
43
+ return enhanced_img,score_niqe,0
44
  else:
45
  return enhanced_img,0,0
46