Spaces:
Runtime error
Runtime error
Alex Chan
commited on
Commit
•
af39e20
1
Parent(s):
f8464e1
lowered confidence interval for mask
Browse files- Code/Inference.py +1 -1
Code/Inference.py
CHANGED
@@ -149,7 +149,7 @@ def Inference(frame,predictor,dlc_liveObj,ScaleBBox=1,Dilate=5,DLCThreshold=0.3)
|
|
149 |
BirdIndex = np.where(outputs.pred_classes.numpy() == 14)[0] #14 is ID for bird
|
150 |
BirdBBox = outputs.pred_boxes[BirdIndex].tensor.numpy()
|
151 |
# import ipdb;ipdb.set_trace()
|
152 |
-
BirdMasks = (outputs.pred_masks>0.
|
153 |
|
154 |
for x in range(BirdBBox.shape[0]):
|
155 |
# import ipdb;ipdb.set_trace()
|
|
|
149 |
BirdIndex = np.where(outputs.pred_classes.numpy() == 14)[0] #14 is ID for bird
|
150 |
BirdBBox = outputs.pred_boxes[BirdIndex].tensor.numpy()
|
151 |
# import ipdb;ipdb.set_trace()
|
152 |
+
BirdMasks = (outputs.pred_masks>0.7).numpy()[BirdIndex]
|
153 |
|
154 |
for x in range(BirdBBox.shape[0]):
|
155 |
# import ipdb;ipdb.set_trace()
|