turhancan97 commited on
Commit
5f60f25
·
1 Parent(s): c53d901

app deploy

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -102,7 +102,7 @@ def show_preds_image(image_path):
102
  width=img_res_toshow.shape[1]
103
  dim=(width,height)
104
 
105
- outputs = model.predict(source=image_path)
106
 
107
  #obtain BW image
108
  bw=(outputs[0].masks.masks[0].cpu().numpy() * 255).astype("uint8")
@@ -130,7 +130,7 @@ def show_preds_image(image_path):
130
  img,
131
  (int(det[0]), int(det[1])),
132
  (int(det[2]), int(det[3])),
133
- color=(0, 0, 255),
134
  thickness=2,
135
  lineType=cv2.LINE_AA
136
  )
 
102
  width=img_res_toshow.shape[1]
103
  dim=(width,height)
104
 
105
+ outputs = model.predict(source=img_res_toshow)
106
 
107
  #obtain BW image
108
  bw=(outputs[0].masks.masks[0].cpu().numpy() * 255).astype("uint8")
 
130
  img,
131
  (int(det[0]), int(det[1])),
132
  (int(det[2]), int(det[3])),
133
+ color=(255, 0, 0),
134
  thickness=2,
135
  lineType=cv2.LINE_AA
136
  )