Jacopo commited on
Commit
13c9775
1 Parent(s): 09ef5ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -48,13 +48,13 @@ def nogan(input_img):
48
  i = np.transpose(i, (2, 0, 1))
49
  i = np.expand_dims(i, 0)
50
  i = i / 255.0
51
- i = normalize(i, (0.485, 0.456, 0.406), (0.229, 0.224, 0.225))
52
 
53
  ort_outs = ort_session.run([output_name], {input_name: i})
54
  output = ort_outs
55
  output = output[0][0]
56
 
57
- output = denormalize(output, (0.485, 0.456, 0.406), (0.229, 0.224, 0.225))
58
  output = output * 255.0
59
  output = output.astype('uint8')
60
  output = np.transpose(output, (1, 2, 0))
 
48
  i = np.transpose(i, (2, 0, 1))
49
  i = np.expand_dims(i, 0)
50
  i = i / 255.0
51
+ #i = normalize(i, (0.485, 0.456, 0.406), (0.229, 0.224, 0.225))
52
 
53
  ort_outs = ort_session.run([output_name], {input_name: i})
54
  output = ort_outs
55
  output = output[0][0]
56
 
57
+ #output = denormalize(output, (0.485, 0.456, 0.406), (0.229, 0.224, 0.225))
58
  output = output * 255.0
59
  output = output.astype('uint8')
60
  output = np.transpose(output, (1, 2, 0))