turhancan97 commited on
Commit
8493ae9
1 Parent(s): f53de6f

add conf parameter to video prediction

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -175,7 +175,7 @@ def show_preds_video(video_path):
175
  height=img_res_toshow.shape[0]
176
  width=img_res_toshow.shape[1]
177
  dim=(width,height)
178
- outputs = model.predict(source=img_res_toshow)
179
  for object in range(len(outputs[0].masks.masks)):
180
  #obtain BW image
181
  bw=(outputs[0].masks.masks[object].cpu().numpy() * 255).astype("uint8")
 
175
  height=img_res_toshow.shape[0]
176
  width=img_res_toshow.shape[1]
177
  dim=(width,height)
178
+ outputs = model.predict(source=img_res_toshow,conf=0.4)
179
  for object in range(len(outputs[0].masks.masks)):
180
  #obtain BW image
181
  bw=(outputs[0].masks.masks[object].cpu().numpy() * 255).astype("uint8")