Update functions.py

#1
by chitversion - opened
Files changed (1) hide show
  1. functions.py +1 -1
functions.py CHANGED
@@ -25,7 +25,7 @@ def predict(inp_image):
25
  output_name = session.get_outputs()[0].name
26
 
27
  pred_onx = session.run([output_name], {input_name: image.astype(np.float32)})[0]
28
- pred_onx = pred_onx > 0.5
29
  pred_onx = pred_onx * 255
30
 
31
  pred_onx = cv2.resize(pred_onx[0, 0].astype(np.uint8) , (inp_dim[1], inp_dim[0]))
 
25
  output_name = session.get_outputs()[0].name
26
 
27
  pred_onx = session.run([output_name], {input_name: image.astype(np.float32)})[0]
28
+ pred_onx = pred_onx > 0.3
29
  pred_onx = pred_onx * 255
30
 
31
  pred_onx = cv2.resize(pred_onx[0, 0].astype(np.uint8) , (inp_dim[1], inp_dim[0]))