Spaces:
Running
Running
turhancan97
commited on
Commit
·
8493ae9
1
Parent(s):
f53de6f
add conf parameter to video prediction
Browse files
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")
|