Spaces:
Sleeping
Sleeping
Update python_utils/get_model.py
Browse files
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.
|
| 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 |
|