Alex Chan commited on
Commit
af39e20
1 Parent(s): f8464e1

lowered confidence interval for mask

Browse files
Files changed (1) hide show
  1. 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.95).numpy()[BirdIndex]
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()