jatamura commited on
Commit
0f64244
·
verified ·
1 Parent(s): cfd1ae2

Update python_utils/get_model.py

Browse files
Files changed (1) hide show
  1. python_utils/get_model.py +1 -1
python_utils/get_model.py CHANGED
@@ -63,7 +63,7 @@ def mask_nms(masks, scores, nms_threshold=0.5):
63
  order.remove(j)
64
  return keep
65
 
66
- def apply_nms(prediction, cls_agnostic_nms=0.5, mask=False):
67
  from torchvision.ops import nms
68
  from detectron2.structures import Instances
69
 
 
63
  order.remove(j)
64
  return keep
65
 
66
+ def apply_nms(prediction, cls_agnostic_nms=0.85, mask=False):
67
  from torchvision.ops import nms
68
  from detectron2.structures import Instances
69