Update functions.py
#1
by
chitversion
- opened
- 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.
|
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]))
|